最大权独立集的练习题
C++ 代码
#include<iostream>
#include<cstring>
using namespace std;
const int N=1010,M=200010,INF=2e9;
int n,m,S,T;
int h[N],e[M],f[M],ne[M],idx;
int q[N],d[N],cur[N];
int cost[50][50];
int dx[]={-1,0,1,0},dy[]={0,1,0,-1};
int get(int x,int y)
{
return (x-1)*m+y;
}
void add(int a,int b,int c)
{
e[idx]=b,f[idx]=c,ne[idx]=h[a],h[a]=idx++;
e[idx]=a,f[idx]=0,ne[idx]=h[b],h[b]=idx++;
}
bool bfs()
{
memset(d,-1,sizeof d);
int hh=0,tt=0;
q[0]=S,d[S]=0,cur[S]=h[S];
while(hh<=tt)
{
int t=q[hh++];
for(int i=h[t];~i;i=ne[i])
{
int ver=e[i];
if(d[ver]==-1&&f[i])
{
d[ver]=d[t]+1;
cur[ver]=h[ver];
if(ver==T) return true;
q[++tt]=ver;
}
}
}
return false;
}
int find(int u,int limit)
{
if(u==T) return limit;
int flow=0;
for(int i=cur[u];~i&&flow<limit;i=ne[i])
{
cur[u]=i;
int ver=e[i];
if(d[ver]==d[u]+1&&f[i])
{
int t=find(ver,min(f[i],limit-flow));
if(!t) d[ver]=-1;
f[i]-=t,f[i^1]+=t,flow+=t;
}
}
return flow;
}
int dinic()
{
int res=0,flow;
while(bfs()) while(flow=find(S,INF)) res+=flow;
return res;
}
int main()
{
memset(h,-1,sizeof h);
scanf("%d%d",&n,&m);
S=0,T=n*m+1;
int tot=0;
for(int i=1;i<=n;++i)
for(int j=1;j<=m;++j)
scanf("%d",&cost[i][j]),tot+=cost[i][j];
for(int i=1;i<=n;++i)
for(int j=1;j<=m;++j)
{
if(i+j&1)
{
add(S,get(i,j),cost[i][j]);
for(int k=0;k<4;++k)
{
int x=i+dx[k],y=j+dy[k];
if(x>=1&&x<=n&&y>=1&&y<=m)
add(get(i,j),get(x,y),INF);
}
}
else add(get(i,j),T,cost[i][j]);
}
printf("%d",tot-dinic());
return 0;
}