最大点权独立集 网络流sap解法

给你一个n*n的格子的棋盘,每个格子里面有一个非负数。
从中取出若干个数,使得任意的两个数所在的格子没有公共边,就是说所取的数所在的2个格子不能相邻,并且取出的数的和最大。

#include<bits/stdc++.h>
using namespace std;
const int maxn=100;
const int maxm=maxn*maxn;
const int inf=0x3f3f3f3f;
int head[maxm],cnt=0;
int mapp[maxn][maxn];
struct edge
{
    int v,nxt,w;
}edge[maxm*3+100];
int n,m;
int dir[4][2]={0,1,1,0,0,-1,-1,0};
void add_edge(int u,int v,int w)
{
    edge[cnt].v=v;
    edge[cnt].w=w;
    edge[cnt].nxt=head[u];
    head[u]=cnt++;
    //edge[cnt].u=v;
    edge[cnt].v=u;
    edge[cnt].w=0;
    //edge[cnt].cost=-cost;
    edge[cnt].nxt=head[v];
    head[v]=cnt++;
}
int numh[maxm],h[maxm],curedge[maxm],pre[maxm];
int sap(int s,int t)
{
    memset(numh,0,sizeof(numh));
    memset(h,0,sizeof(h));
    memset(pre,-1,sizeof(pre));
    int cur_flow,flow_ans=0,u,tmp,neck,i;
    for(i=1;i<=n;i++)
    {
        curedge[i]=head[i];
    }
    numh[0]=n;
    u=s;
    while(h[s]<n)
    {
        //printf("%d\n",h[s]);
        //printf("1111\n");
        if(u==t)
        {
            cur_flow=inf;
            for(i=s;i!=t;i=edge[curedge[i]].v)
            {
                if(cur_flow>edge[curedge[i]].w)
                {
                    neck=i;
                    cur_flow=edge[curedge[i]].w;
                }
            }
            for(i=s;i!=t;i=edge[curedge[i]].v)
            {
                tmp=curedge[i];
                edge[tmp].w-=cur_flow;
                edge[tmp^1].w+=cur_flow;
            }
            flow_ans+=cur_flow;
            u=neck;
        }
        for(i=curedge[u];i!=-1;i=edge[i].nxt)
        {
            if(edge[i].w&&h[u]==h[edge[i].v]+1)
            {
                break;
            }
        }
        if(i!=-1)
        {
            curedge[u]=i;
            pre[edge[i].v]=u;
            u=edge[i].v;
        }
        else
        {
            if(0==--numh[h[u]])
                break;
            curedge[u]=head[u];
            for(tmp=n,i=head[u];i!=-1;i=edge[i].nxt)
            {
                if(edge[i].w)
                {
                    tmp=min(tmp,h[edge[i].v]);
                }
            }
            h[u]=tmp+1;
            ++numh[h[u]];
            if(u!=s)
                u=pre[u];
        }
    }
    return flow_ans;
}
int main ()
{
    int m;

    while(~scanf("%d",&m))
    {
        int sum=0;
        for(int i=1;i<=m;i++)
        {
            for(int j=1;j<=m;j++)
            {
                scanf("%d",&mapp[i][j]);
                sum+=mapp[i][j];
            }
        }
        memset(head,-1,sizeof(head));
        cnt=0;
        for(int i=1;i<=m;i++)
        {
            for(int j=1;j<=m;j++)
            {
                if((i+j)%2==0)
                {
                    add_edge(m*m+1,((i-1)*m+j),mapp[i][j]);
                }
                else
                {
                    add_edge(((i-1)*m+j),m*m+2,mapp[i][j]);
                }
            }
        }
        for(int i=1;i<=m;i++)
        {
            for(int j=1;j<=m;j++)
            {
                if((i+j)%2==0)
                {
                    for(int t=0;t<4;t++)
                    {
                        int x,y;
                        x=i+dir[t][0];
                        y=j+dir[t][1];
                        if(x>=1&&y>=1&&x<=m&&y<=m)
                        {
                            add_edge(((i-1)*m+j),((x-1)*m+y),inf);
                        }
                    }
                }   
            }
        }
        n=m*m+2;
        int answer=sap(m*m+1,m*m+2);
        printf("%d\n",sum-answer); 
    }

}

猜你喜欢

转载自blog.csdn.net/qq_36616023/article/details/80251601
今日推荐