一上来容易想到缩点dp,但由于一个集合是没法用状态表示的,所以不能dp
然后就可以想到最小割,即通过割边和代价完成分配;
就比较好做了,默认选0就和0连边,默认选1就和1连边,朋友之间连边,如果在同一个集合里不会产生流量,如果在不同集合里会产生代价为1的流量
码:
#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;
#define N 900005
#define inf 1000000009
queue<int>q;
int tot=-1,hou[N],xia[N],zhong[N],yuan[N],v[N],d[N],s,t,n,m,x,y,i;
void jian(int a,int b,int c)
{
++tot;hou[tot]=yuan[a],zhong[tot]=b,yuan[a]=tot,v[tot]=c;
}
void jia(int a,int b,int c)
{
jian(a,b,c);
jian(b,a,0);
}
bool bfs()
{
int i,j;
for(i=1;i<=t;i++)
{
xia[i]=yuan[i];d[i]=inf;
}
d[s]=0;
q.push(s);
while(!q.empty())
{
int st=q.front();
q.pop();
for(i=xia[st];i!=-1;i=hou[i])
{ if(v[i]==0)continue;
int nd=zhong[i];
if(d[nd]==inf)d[nd]=d[st]+1,q.push(nd);
}
}
return d[t]!=inf;
}
int dfs(int o,int limit)
{
if(o==t||limit==0)return limit;
int i,f,flow=0;
for( i=xia[o];i!=-1;i=hou[i])
{
xia[o]=i;
int nd=zhong[i];
if(d[nd]==d[o]+1&&(f=dfs(nd,min(limit,v[i]))))
{
flow+=f;
limit-=f;
v[i]-=f;
v[i^1]+=f;
if(limit==0)break;
}
}
return flow;
}
int dinic()
{
int ans=0;
while(bfs())
{
ans+=dfs(s,inf);
} return ans;
}
int main()
{
memset(yuan,-1,sizeof(yuan));
scanf("%d%d",&n,&m);
s=n+1;t=n+2;
for(i=1;i<=n;i++)
{
scanf("%d",&x);
if(x==1)jia(i,t,1);
else jia(s,i,1);
}
for(i=1;i<=m;i++)
scanf("%d%d",&x,&y),jia(x,y,1),jia(y,x,1);
printf("%d",dinic());
}