给出起点和终点,求最小割,也就是起点到终点的最大流
把每个城市拆成入点和出点,中间连边,边权为花费
其他边就连入时就接入点,连出时接出点,边权为INF
题目说可以在任意城市设立岗哨,也包括起点s和终点t
所以新的起点为s的入点,终点为d的出点,用ISAP走一遍就行了
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
const int N=500+10;
const int M=1e5+10;
const int INF=0x7f7f7f7f;
struct Edge
{
int to,nxt,cap,flow;
}edge[M];
int tot,first[N];
void addedge(int u,int v,int w,int rw=0)
{
edge[tot].to=v;edge[tot].cap=w;edge[tot].flow=0;
edge[tot].nxt=first[u];first[u]=tot++;
edge[tot].to=u;edge[tot].cap=rw;edge[tot].flow=0;
edge[tot].nxt=first[v];first[v]=tot++;
}
void init()
{
tot=0;
memset(first,-1,sizeof(first));
}
int gap[N],dep[N],cur[N];
int Q[N],S[N];
void bfs(int s)
{
memset(gap,0,sizeof(gap));
memset(dep,-1,sizeof(dep));
int q0=0,q1=0;
Q[q1++]=s;
dep[s]=0;
gap[0]=1;
while(q0<q1)
{
int u=Q[q0++];
for(int i=first[u];i!=-1;i=edge[i].nxt)
{
int v=edge[i].to;
if(dep[v]!=-1) continue;
dep[v]=dep[u]+1;
gap[dep[v]]++;
Q[q1++]=v;
}
}
}
int sap(int s,int t,int n)
{
int ans=0;
bfs(t);
memcpy(cur,first,sizeof(first));
int top=0;
int u=s;
while(dep[s]<n)
{
if(u==t)
{
int Min=INF;
int inser;
for(int i=0;i<top;i++)
if(Min>edge[S[i]].cap-edge[S[i]].flow)
{
Min=edge[S[i]].cap-edge[S[i]].flow;
inser=i;
}
for(int i=0;i<top;i++)
{
edge[S[i]].flow+=Min;
edge[S[i]^1].flow-=Min;
}
ans+=Min;
top=inser;
u=edge[S[top]^1].to;
}
bool flag=false;
int v;
for(int i=cur[u];i!=-1;i=edge[i].nxt)
{
v=edge[i].to;
if(edge[i].cap-edge[i].flow&&dep[v]+1==dep[u])
{
flag=true;
cur[u]=S[top++]=i;
break;
}
}
if(flag)
{
u=v;
continue;
}
gap[dep[u]]--;
if(!gap[dep[u]]) return ans;
int Min=n;
for(int i=first[u];i!=-1;i=edge[i].nxt)
if(edge[i].cap-edge[i].flow&&Min>dep[edge[i].to])
{
Min=dep[edge[i].to];
cur[u]=i;
}
dep[u]=Min+1;
gap[dep[u]]++;
if(u!=s) u=edge[S[--top]^1].to;
}
return ans;
}
int main()
{
int n,m,s,t;
while(~scanf("%d%d",&n,&m))
{
scanf("%d%d",&s,&t);
init();
int a,b;
for(int i=1;i<=n;i++)
{
scanf("%d",&a);
addedge(i,i+n,a);
}
for(int i=0;i<m;i++)
{
scanf("%d%d",&a,&b);
addedge(a+n,b,INF);
addedge(b+n,a,INF);
}
printf("%d\n",sap(s,t+n,2*n));
}
return 0;
}