P2057 [SHOI2007] 善意的投票(最小割)

Posted 2022-06-02 Harris-H

P2057 [SHOI2007] 善意的投票(最小割)

同意与非同意 建立S-T。

然后S 连同意的人，T连不同意的人。

好朋友之间连双向边。

连双向边的原因是关系之间是双向，且可能发生同阵营的人之间是好朋友。

然后跑最大流最小割即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull; 
const int N=305,M=2e5+5,inf=0x3f3f3f3f,mod=1e9+7;
const int hashmod[4] = 402653189,805306457,1610612741,998244353;
#define mst(a,b) memset(a,b,sizeof a)
#define db double
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define x first
#define y second
#define pb emplace_back
#define SZ(a) (int)a.size()
#define rep(i,a,b) for(int i=a;i<=b;++i)
#define per(i,a,b) for(int i=a;i>=b;--i)
#define ios ios::sync_with_stdio(false),cin.tie(nullptr) 
void Print(int *a,int n)
	for(int i=1;i<n;i++)
		printf("%d ",a[i]);
	printf("%d\\n",a[n]); 

template <typename T>		//x=max(x,y)  x=min(x,y)
void cmx(T &x,T y)
	if(x<y) x=y;

template <typename T>
void cmn(T &x,T y)
	if(x>y) x=y;

struct Dinic
//Dinic O(n^2m)
	int n,m,st,ed;
	int id(int x,int y)
		return (x-1)*m+y;
	
	struct edge
		int to,nt;
		ll w;
	e[M];
	int h[N],cur[N],cnt,dep[N];
	void init(int _st,int _ed)
		st=_st,ed=_ed;
		cnt=1;mst(h,0);
	
	Dinic(int _st=0,int _ed=0)init(_st,_ed);
	void add(int u,int v,ll w)
		e[++cnt]=v,h[u],w,h[u]=cnt;
		e[++cnt]=u,h[v],0,h[v]=cnt;
	
	ll dfs(int u,ll c)	//search for augment path
		if(u==ed) return c;
		ll res=c;
		for(int &i=cur[u];i;i=e[i].nt)
			int v=e[i].to; ll w=e[i].w;
			if(w&&dep[v]==dep[u]+1)
				ll now=dfs(v,min(res,w));
				if(!now) dep[v]=1;
				else e[i].w-=now,e[i^1].w+=now,res-=now;
			
			if(!res) return c;
		return c-res;
	 
	bool bfs()		//layer the graph
		queue<int>q;q.push(st);mst(dep,0),dep[st]=1;
		while(!q.empty())
			int u=q.front();q.pop();cur[u]=h[u];
			for(int i=h[u];i;i=e[i].nt)
				int v=e[i].to;ll w=e[i].w;
				if(w&&!dep[v]) dep[v]=dep[u]+1,q.push(v);
			
		return dep[ed];
	 
	ll dinic()
		ll s=0;
		while(bfs()) s+=dfs(st,inf);
		return s;
	
G;
int main()
	int n,m;
	scanf("%d%d",&n,&m);
	int st = 0 ,ed = n+1;
	G.init(0,n+1);
	rep(i,1,n)
		int x;scanf("%d",&x);
		if(x) G.add(st,i,1);
		else G.add(i,ed,1);
	
	while(m--)
		int u,v;
		scanf("%d%d",&u,&v);
		G.add(u,v,1);
		G.add(v,u,1);
	
	printf("%lld\\n",G.dinic());
	return 0;