题目链接:http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=26869
思路:题目还是比较水的,由于点也有容量,则必须拆点,然后跑最大流。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
#define MAXN 222
#define MAXM 222222
#define inf 1<<30
#define FILL(a,b) memset(a,b,sizeof(a)) struct Edge{
int v,cap,next;
}edge[MAXM]; int n,m,vs,vt,NV,NE,Cap[MAXN];
int head[MAXN]; void Insert(int u,int v,int cap)
{
edge[NE].v=v;
edge[NE].cap=cap;
edge[NE].next=head[u];
head[u]=NE++; edge[NE].v=u;
edge[NE].cap=;
edge[NE].next=head[v];
head[v]=NE++;
} int level[MAXN],gap[MAXN];
void bfs(int vt)
{
FILL(level,-);
FILL(gap,);
queue<int>que;
que.push(vt);
level[vt]=;
gap[]++;
while(!que.empty()){
int u=que.front();
que.pop();
for(int i=head[u];i!=-;i=edge[i].next){
int v=edge[i].v;
if(level[v]!=-)continue;
level[v]=level[u]+;
gap[level[v]]++;
que.push(v);
}
}
} int pre[MAXN],cur[MAXN];
int SAP(int vs,int vt)
{
bfs(vt);
memcpy(cur,head,sizeof(head));
int u=pre[vs]=vs,aug=inf,maxflow=;
gap[]=NV;
while(level[vs]<NV){
bool flag=false;
for(int &i=cur[u];i!=-;i=edge[i].next){
int v=edge[i].v;
if(edge[i].cap>&&level[u]==level[v]+){
flag=true;
aug=min(aug,edge[i].cap);
pre[v]=u;
u=v;
if(v==vt){
maxflow+=aug;
for(u=pre[u];v!=vs;v=u,u=pre[u]){
edge[cur[u]].cap-=aug;
edge[cur[u]^].cap+=aug;
}
aug=inf;
}
break;
}
}
if(flag)continue;
int minlevel=NV;
for(int i=head[u];i!=-;i=edge[i].next){
int v=edge[i].v;
if(edge[i].cap>&&level[v]<minlevel){
minlevel=level[v];
cur[u]=i;
}
}
if(--gap[level[u]]==)break;
level[u]=minlevel+;
gap[level[u]]++;
u=pre[u];
}
return maxflow;
} int main()
{
int _case,u,v,k1,k2,cap,t=;
scanf("%d",&_case);
while(_case--){
scanf("%d",&n);
NE=;
FILL(head,-);
vs=,vt=*n+,NV=vt+;
for(int i=;i<=n;i++)scanf("%d",&Cap[i]);
scanf("%d",&m);
while(m--){
scanf("%d%d%d",&u,&v,&cap);
Insert(u+n,v,cap);
}
for(int i=;i<=n;i++)Insert(i,i+n,Cap[i]);
scanf("%d%d",&k1,&k2);
while(k1--){
scanf("%d",&u);
Insert(vs,u,inf);
}
while(k2--){
scanf("%d",&u);
Insert(u+n,vt,inf);
}
printf("Case %d: %d\n",t++,SAP(vs,vt));
}
return ;
}