洛谷P2057 [SHOI2007]善意的投票 题解

题目链接：

https://www.luogu.org/problemnew/show/P2057

分析：

由0和1的选择我们直觉的想到0与S一堆，1与T一堆。

但是发现，刚开始的主意并不一定是最终的结果。

于是用源点S表示最终选择0的集合。

汇点T表示最终选择1的集合。

如果一个人P选择了0，那么$S->P$连一条流量为1的边，然后$P->T$连一条流量为0的边。

反之如果P选择了1，那么$S->P$连一条流量为0的边，然后$P->T$连一条流量为1的边。

最后把所有朋友对之间连流量为1的双向边（关系相互）

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<queue>
#define inf 0x7fffffff
using namespace std;
int s,t,ans;
int d[505];
struct edge
{
	int to,val,rev;
	edge(int _to,int _val,int _rev)
	{
		to=_to;
		val=_val;
		rev=_rev;
	}
};
vector<edge>e[505];
void add(int x,int y,int w)
{
	e[x].push_back(edge(y,w,e[y].size()));
	e[y].push_back(edge(x,0,e[x].size()-1));
}
bool bfs()
{
	queue<int>q; 
	memset(d,-1,sizeof(d));
	d[s]=0;
	q.push(s);
	while(!q.empty())
	{
		int x=q.front();q.pop();
		for(int i=0;i<e[x].size();i++)
		{
			int y=e[x][i].to;
			if(d[y]==-1&&e[x][i].val)
			{
				q.push(y);
				d[y]=d[x]+1;
			}
		}
	} 
	if(d[t]==-1)return 0;
	return 1;
}
int dfs(int x,int low) 
{
    if(x==t || low==0)
        return low;
    int totflow=0; 
    for(int i=0;i<e[x].size();i++)
    {
        int y=e[x][i].to;
        int rev=e[x][i].rev;
        if(d[y]==d[x]+1 && e[x][i].val)
        {
            int a=dfs(y,min(low,e[x][i].val)); 
            e[x][i].val-=a;
            e[y][rev].val+=a;
            low-=a;
            totflow+=a;
            if(low==0) 
                return totflow;
        }
    }
    if(low!=0) 
        d[x]=-1;
    return totflow;
}

void dinic()
{
	while(bfs())
	{
		//printf("%d
",ans); 
		ans+=dfs(s,inf); 
	}
}
int main()
{
	int n,m,x,y,o;
	scanf("%d%d",&n,&m);
	s=0;t=500;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&o);
		if(o==0)
		{
			add(s,i,1);//add(i,t,0);
		}
		else
		{
			//add(s,i,0);
			add(i,t,1);
		}
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y,1);add(y,x,1);
	}
	dinic();
	printf("%d
",ans);
	return 0;
}