1. 程式人生 > >BZOJ 3511: 土地劃分

BZOJ 3511: 土地劃分

題意:Y國有N座城市,並且有M條雙向公路將這些城市連線起來,並且任意兩個城市至少有一條路徑可以互達。
Y國的國王去世之後,他的兩個兒子A和B都想成為新的國王,但他們都想讓這個國家更加安定,不會用武力解決問題。
於是他們想將這個國家分成兩個小國家A國和B國。現在,A擁有1號城市,B擁有N號城市,其他的城市還尚未確定歸屬哪邊(劃分之後的國家內部城市可以不連通)。
由於大家都想讓國家變得更好,而某些城市的人民願意國王的A兒子作為他們的領袖,而某些城市更看好B,而為了交通的便捷,如果劃分後的公路連線兩個同一個國家的城市,那麼更利於城市之間的交流。於是大臣們設計了一種對土地劃分的評分機制,具體如下:
1. 對於城市i,如果它劃分給A國,將得到VA[i]的得分;劃分給B國,將得到VB[i]的得分。
2. 對於一條公路i,如果它連線兩個A國的城市,將得到EA[i]的得分;連線兩個B國的城市,將得到EB[i]的得分;否則,這條公路將失去意義,將扣除EC[i]的得分。
現請你找到最優的土地劃分,使得這種它的評分最高。

很明顯的二元關係的最小割,新建S,T,對於每個點,由S向I連VA[I],由I向T連VB[I].對於每條邊,由S向x,y連EA/2,由T向x,y連EB/2.x,y互連EA/2+EB/2+EC,特別地,S向1連INF,N向T連INF.我們先將所有可能的收益加起來,網路流中的邊權代表扣除的收益,這樣跑最小割之後就是剩餘的最大收益。
Tips:先將所有權值乘上2,最後除以2,這樣避免實數運算。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector>
#include<queue> using namespace std; const int maxn=10000+10; const int INF=1000000000; int n,m,s,t,st[maxn],ed[maxn],ans,cur[maxn],vis[maxn],dis[maxn]; struct edge { int from,to,cap,flow; }; vector<int> g[maxn]; vector<edge> edges; void addedge(int from,int to,int cap) { edges.push_back((edge){from,to,cap,0
}); edges.push_back((edge){to,from,0,0}); int m1=edges.size(); g[from].push_back(m1-2); g[to].push_back(m1-1); } bool bfs() { memset(vis,0,sizeof(vis)); queue<int> Q;Q.push(s);vis[s]=1;dis[s]=0; while(!Q.empty()) { int x=Q.front();Q.pop(); for(int i=0;i<g[x].size();i++) { edge e=edges[g[x][i]]; if(e.cap>e.flow&&!vis[e.to]) { vis[e.to]=1; dis[e.to]=dis[x]+1; Q.push(e.to); } } } return vis[t]; } int dfs(int x,int a) { if(x==t||a==0) return a; int flow=0,f; for(int &i=cur[x];i<g[x].size();i++) { edge &e=edges[g[x][i]]; if(dis[e.to]==dis[x]+1&&(f=dfs(e.to,min(a,e.cap-e.flow)))>0) { flow+=f; e.flow+=f; edges[g[x][i]^1].flow-=f; a-=f; if(a==0) break; } } return flow; } int maxflow() { int flow=0; while(bfs()) { memset(cur,0,sizeof(cur)); flow+=dfs(s,INF); } return flow; } int main() { //freopen("3511.in","r",stdin); //freopen("3511.out","w",stdout); scanf("%d%d",&n,&m);s=0,t=n+1;st[1]=INF;ed[n]=INF; for(int i=2;i<=n-1;i++) { int x;scanf("%d",&x); st[i]+=2*x; ans+=2*x; } for(int i=2;i<=n-1;i++) { int x;scanf("%d",&x); ed[i]+=2*x;ans+=2*x; } for(int i=1;i<=m;i++) { int x,y,ea,eb,ec; scanf("%d%d%d%d%d",&x,&y,&ea,&eb,&ec); ans+=(ea+eb)*2;st[x]+=ea;st[y]+=ea;ed[x]+=eb;ed[y]+=eb; addedge(x,y,ea+eb+2*ec);addedge(y,x,ea+eb+2*ec); } for(int i=1;i<=n;i++) { addedge(s,i,st[i]); addedge(i,t,ed[i]); } printf("%d\n",(ans-maxflow())/2); return 0; }