Battle HDU

来源:互联网 发布:对对子软件 编辑:程序博客网 时间:2024/04/30 15:57

由于小白同学近期习武十分刻苦,很快被晋升为天策军的统帅。而他上任的第一天,就面对了一场极其困难的战斗:
据侦查兵回报,前方共有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  0x3f3f3f3fusing 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;}
原创粉丝点击