BZOJ 3511: 土地劃分
阿新 • • 發佈:2019-02-17
題意: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;
}