BZOJ 2768[JLOI2010]冠军调查

题面

题意

n个人发表意见,每个人说1或0,有m对朋友,为了每个人可以为了朋友迁就自己,改变主意,求改变意见次数与和朋友意见不同的组数之和的最小值.

做法

用最小割求解,将超级源点与说1的相连,说0的与超级汇点相连,朋友间连双向边,求最小割.
因为根据题意只有两种操作会是答案变大:
1.与朋友意见不同(去掉一条边).
2.改变意见(去掉与超级源点或超级汇点间的边)
因此可以用最小割来做

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define INF 0x3f3f3f3f
#define N 10100
using namespace std;

int n,m,bb,need[30][30],val[30][30],first[N],deep[N],tmp,s,t,dir[4][2]= {1,0,-1,0,0,-1,0,1},sum,ans,cur[N];
struct Bn
{
    int next,to,quan;
} bn[200100];
queue<int>que;

inline void add(int u,int v,int w)
{
    bn[bb].to=v;
    bn[bb].quan=w;
    bn[bb].next=first[u];
    first[u]=bb;
    bb++;
}

inline void ad(int u,int v,int w)
{
    add(u,v,w);
    add(v,u,0);
}

inline bool bfs()
{
    int p,q;
    for(;!que.empty();que.pop());
    memset(deep,0,sizeof(deep));
    deep[s]=1;
    que.push(s);
    for(;!que.empty();)
    {
        q=que.front();
        que.pop();
        for(p=first[q];p!=-1&&!deep[t];p=bn[p].next)
        {
            if(deep[bn[p].to]||!bn[p].quan) continue;
            deep[bn[p].to]=deep[q]+1;
            que.push(bn[p].to);
        }
    }
    return deep[t];
}

int dfs(int now,int mn)
{
    if(now==t)
    {
        return mn;
    }
    int res;
    for(int &p=cur[now];p!=-1;p=bn[p].next)
    {
        if(deep[bn[p].to]!=deep[now]+1||!bn[p].quan) continue;
        res=dfs(bn[p].to,min(bn[p].quan,mn));
        if(res)
        {
            bn[p].quan-=res;
            bn[p^1].quan+=res;
            return res;
        }
    }
    return 0;
}

int main()
{
    memset(first,-1,sizeof(first));
    register int i,j,p,q,k;
    cin>>n>>m;
    t=n+1;
    for(i=1;i<=n;i++)
    {
        scanf("%d",&p);
        p?ad(s,i,1):ad(i,t,1);
    }
    for(i=1;i<=m;i++)
    {
        scanf("%d%d",&p,&q);
        ad(p,q,1);
        ad(q,p,1);
    }
    for(;bfs();)
    {
        for(i=s;i<=t;i++) cur[i]=first[i];
        for(k=dfs(s,INF);k;ans+=k,k=dfs(s,INF));
    }
    cout<<ans;
}