Kejin Game UVALive - 7264 拆点+最小割

//ISAP int
#include<bits/stdc++.h>
#define ll long long
#define MAXN 1010
#define inf 0x3f3f3f3f
using namespace std;
int n,m;//题目输入点数，边数
struct Edge{
int v,next;
int cap,flow;
}edge[MAXN*100];
int cur[MAXN],pre[MAXN],gap[MAXN],path[MAXN],dep[MAXN];
int cnt=0;//实际存储总边数
void init()
{
cnt=0;
memset(pre,-1,sizeof(pre));
}
void add(int u,int v,int w)//加边
{
edge[cnt].v=v;
edge[cnt].cap=w;
edge[cnt].flow=0;
edge[cnt].next=pre[u];
pre[u]=cnt++;
}
bool bfs(int s,int t)//其实这个bfs可以融合到下面的迭代里，但是好像是时间要长
{
memset(dep,-1,sizeof(dep));
memset(gap,0,sizeof(gap));
gap[0]=1;
dep[t]=0;
queue<int>q;
while(!q.empty())
q.pop();
q.push(t);//从汇点开始反向建层次图
while(!q.empty())
{
int u=q.front();
q.pop();
for(int i=pre[u];i!=-1;i=edge[i].next)
{
int v=edge[i].v;
if(dep[v]==-1&&edge[i^1].cap>edge[i^1].flow)//注意是从汇点反向bfs，但应该判断正向弧的余量
{
dep[v]=dep[u]+1;
gap[dep[v]]++;
q.push(v);
//if(v==s)//感觉这两句优化加了一般没错，但是有的题可能会错，所以还是注释出来，到时候视情况而定
//break;
}
}
}
return dep[s]!=-1;
}
int isap(int s,int t)
{
bfs(s,t);
memcpy(cur,pre,sizeof(pre));
int u=s;
path[u]=-1;
int ans=0;
while(dep[s]<n)//迭代寻找增广路
{
if(u==t)
{
int f=inf;
for(int i=path[u];i!=-1;i=path[edge[i^1].v])//修改找到的增广路
f=min(f,edge[i].cap-edge[i].flow);
for(int i=path[u];i!=-1;i=path[edge[i^1].v])
{
edge[i].flow+=f;
edge[i^1].flow-=f;
}
ans+=f;
u=s;
continue;
}
bool flag=false;
int v;
for(int i=cur[u];i!=-1;i=edge[i].next)
{
v=edge[i].v;
if(dep[v]+1==dep[u]&&edge[i].cap-edge[i].flow)
{
cur[u]=path[v]=i;//当前弧优化
flag=true;
break;
}
}
if(flag)
{
u=v;
continue;
}
int x=n;
if(!(--gap[dep[u]]))return ans;//gap优化
for(int i=pre[u];i!=-1;i=edge[i].next)
{
if(edge[i].cap-edge[i].flow&&dep[edge[i].v]<x)
{
x=dep[edge[i].v];
cur[u]=i;//常数优化
}
}
dep[u]=x+1;
gap[dep[u]]++;
if(u!=s)//当前点没有增广路则后退一个点
u=edge[path[u]^1].v;
}
return ans;
}
int main()
{
//std::ios::sync_with_stdio(0);
int T,s;
cin>>T;
while(T--)
{
init();
scanf("%d%d%d",&n,&m,&s);
int a,b,c;
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&a,&b,&c);
add(a+n,b,c);
add(b,a+n,0);//注意加反向边时一般初始流量为0！
}
for(int i=1;i<=n;i++)
{
scanf("%d",&a);
add(0,i,a);
add(i,0,0);
}
for(int i=1;i<=n;i++)
{
scanf("%d",&a);
add(i,i+n,a);
add(i+n,i,0);
}
add(s+n,2*n+1,inf);
add(2*n+1,s+n,0);
n=2*n+1;
printf("%d\n",isap(0,n));
}
return 0;
}