由于小白同学近期习武十分刻苦,很快被晋升为天策军的统帅。而他上任的第一天,就面对了一场极其困难的战斗:
据侦查兵回报,前方共有N座城池,考虑到地势原因,最终得到一个结论:攻占某些城池之前必须攻占另外一些城池。
事实上,可以把地图看做是一张拓扑图,而攻占某个城池,就意味着必须先攻占它的所有前驱结点。
小白还做了一份调查,得到了攻占每个城池会对他的兵力产生多少消耗(当然也可能会得到增长,因为每攻占一个城池,便可以整顿军队,扩充兵力,天策军的兵力十分庞大,如果不考虑收益,他们可以攻取所有的城池)。
现在请你帮小白统帅做一份战斗计划,挑选攻打哪些城市,使得天策军在战斗过后军容最为壮大。
Input
首先输入一个N 代表有N个城池(1<= n <= 500)
接着输入一个M,代表城池和城池之间的拓扑关系数。
接着输入N个数字 代表从1 到 N 编号城池的战斗消耗(负数代表将要消耗天策军兵力,正数表示天策军可以获得相应的战斗收益)
最后M行 每行2个数字 a,b,代表相应城池的编号。
表示攻占b之后才可以攻占a;
Output
天策军最大能获得多少战斗收益
Sample Input
5 5
8
-8
-10
12
-10
1 2
2 5
1 4
3 4
4 5
Sample Output
2
关键就是建图了,因为题目说了 ,必须先选了b才能选a,换句话说就是如果选了a就必须选b,所以,建图是a到b的有向图,符合最大权闭合子图的建图方式。套模板
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=505;
const int maxx=251005;
int edge;
int to[maxx],flow[maxx],nex[maxx];
int head[maxn];
void addEdge(int v,int u,int cap)
{
to[edge]=u,flow[edge]=cap,nex[edge]=head[v],head[v]=edge++;
to[edge]=v,flow[edge]=0,nex[edge]=head[u],head[u]=edge++;
}
int vis[maxn];
int pre[maxn];
bool bfs(int s,int e)
{
queue<int> que;
pre[s]=-1;
memset(vis,-1,sizeof(vis));
que.push(s);
vis[s]=0;
while(!que.empty())
{
int u=que.front();
que.pop();
for(int i=head[u];~i;i=nex[i])
{
int v=to[i];
if(vis[v]==-1&&flow[i])
{
vis[v]=vis[u]+1;
if(v==e)
return true;
que.push(v);
}
}
}
return false;
}
int dfs(int s,int t,int f)
{
if(s==t||!f)
return f;
int r=0;
for(int i=head[s];~i;i=nex[i])
{
int v=to[i];
if(vis[v]==vis[s]+1&&flow[i])
{
int d=dfs(v,t,min(f,flow[i]));
if(d>0)
{
flow[i]-=d;
flow[i^1]+=d;
r+=d;
f-=d;
if(!f)
break;
}
}
}
if(!r)
vis[s]=INF;
return r;
}
int maxFlow(int s ,int e)//然后直接调用这个即可
{
int ans=0;
while(bfs(s,e))
ans+=dfs(s,e,INF);
return ans;
}
void init()//记得每次使用前初始化
{
memset(head,-1,sizeof(head));
edge=0;
}
int main()
{
int n,m;
int x,y,w;
while(scanf("%d%d",&n,&m)==2)
{
init();
int total=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&w);
if(w>0)
addEdge(0,i,w),total+=w;
else
addEdge(i,n+1,abs(w));
}
while(m--)
{
scanf("%d%d",&x,&y);
addEdge(x,y,INF);
}
printf("%d\n",total-maxFlow(0,n+1));
}
return 0;
}