题意:有n个城市和m条道路(双向),一伙小偷准备从S城出发到H城盗窃,为了将这伙小偷抓住,需要在这n个城市中的每一个城市安排一定数量的警察(每个城市警察的数量已经给出),但警察不希望在S城或H城遇到小偷.求解总共需要的最少警察数.
思路://最小割,把点拆成I,I+N,中间连的点为费用,保证只被切到一次;如果A点和B点相连,2个点连边,边容量为无穷大,保证不被切到 ,由于不能在S,H抓到,所以S,H点和自己S+N,H+N要连无穷大的边,保证不被割到。。
要有拆点的思想。。根据题目的意思来拆。。还要理解网络流的模板。。这个是重点。。
#include<cstdio>
#include<cstring>
#include <iostream>
#include<algorithm>
#define N 100000
#define M 250005 //这个要够大。。
#define inf 999999
using namespace std;
int s,t,num,adj[N],dis[N],q[N];
struct edge
{
int v,w,pre;
}e[M];
int min(int g,int h)
{
if(g>h)
return h;
else
return g;
}
void insert(int u,int v,int w)
{
e[num].v=v;
e[num].w=w;
e[num].pre=adj[u];
adj[u]=num++;
e[num].v=u;
e[num].w=0;
e[num].pre=adj[v];
adj[v]=num++;
}
int bfs()
{
int i,x,v,tail=0,head=0;
memset(dis,0,sizeof(dis));
dis[s]=1;
q[tail++]=s;
while(head<tail)
{
x=q[head++];
for(i=adj[x];i!=-1;i=e[i].pre)
if(e[i].w&&dis[v=e[i].v]==0)
{
dis[v]=dis[x]+1;
if(v==t)
return 1;
q[tail++]=v;
}
}
return 0;
}
int dfs(int s,int limit)
{
if(s==t)
return limit;
int i,v,tmp,cost=0;
for(i=adj[s];i!=-1;i=e[i].pre)
if(e[i].w&&dis[s]==dis[v=e[i].v]-1)
{
tmp=dfs(v,min(limit-cost,e[i].w));
if(tmp>0)
{
e[i].w-=tmp;
e[i^1].w+=tmp;
cost+=tmp;
if(limit==cost)
break;
}
else dis[v]=-1;
}
return cost;
}
int Dinic()
{
int ans=0;
while(bfs())
ans+=dfs(s,inf);
return ans;
}
int main ()
{
int T,n,m,S,h,x;
int i,a,b;
cin>>T;
while(T--)
{
cin>>n>>m>>S>>h;
memset(adj,-1,sizeof(adj));
num=0;
for(i=1;i<=n;i++)
{
cin>>x;
insert(i,i+n,x);
}
for(i=1;i<=m;i++)
{
cin>>a>>b;
insert(a+n,b,inf);
insert(b+n,a,inf);
}
insert(0,S,inf);
insert(S,S+n,inf);
insert(h,h+n,inf);
insert(h+n,n*2+1,inf);
s=0;
t=n*2+1;
cout<<Dinic()<<endl;
}
return 0;
}