問題設定がややこしい。
https://codeforces.com/contest/1288/problem/F
問題
二部グラフを成す無向グラフが与えられる。
各頂点は、赤・青・無色のいずれかである。
ここで、各辺を赤・青・無色のいずれかに塗ることを考える。
赤・青の辺を1辺作るのにR,Bのコストがかかる。
ここで、
- 赤頂点につながる辺は、青辺より赤辺の方が多い
- 青頂点につながる辺は、赤辺より青辺の方が多い
ようにしたい。
条件を満たす塗り方を達成する最小コストと塗り方の例を答えよ。
解法
グラフが二部グラフなので、赤辺を左から右、青辺を右から左に流れた状態として表現する。
ここで最小コストフローを考える。
左側の赤頂点において赤辺が青辺より多い、という条件は最初の容量1だけコストが負で非常に小さく、それ以降はコスト0で流れる、という形で表現できる。
左側の青頂点や、右側の赤青頂点も似た形で表現できる。
あとはコストが負を保った範囲でできるだけフローを流そう。
template<int NV,class V> class MinCostFlow { public: struct edge { int to; V capacity; V cost; int reve; int id;}; vector<edge> E[NV]; int prev_v[NV], prev_e[NV]; V dist[NV]; void add_edge(int x,int y, V cap, V cost, int id=-1) { E[x].push_back((edge){y,cap,cost,(int)E[y].size(),id}); E[y].push_back((edge){x,0, -cost,(int)E[x].size()-1,-1}); /* rev edge */ } V mincost(int from, int to, ll flow) { V res=0; int i,v; ZERO(prev_v); ZERO(prev_e); while(flow>0) { fill(dist, dist+NV, numeric_limits<V>::max()/2); dist[from]=0; priority_queue<pair<V,int> > Q; Q.push(make_pair(0,from)); while(Q.size()) { V d=-Q.top().first; int cur=Q.top().second; Q.pop(); if(dist[cur]!=d) continue; if(d==numeric_limits<V>::max()/2) break; FOR(i,E[cur].size()) { edge &e=E[cur][i]; if(e.capacity>0 && dist[e.to]>d+e.cost) { dist[e.to]=d+e.cost; prev_v[e.to]=cur; prev_e[e.to]=i; Q.push(make_pair(-dist[e.to],e.to)); } } } if(dist[to]>=0) return 1LL<<60; ll lc=flow; for(v=to;v!=from;v=prev_v[v]) lc = min(lc, E[prev_v[v]][prev_e[v]].capacity); flow -= lc; res += lc*dist[to]; for(v=to;v!=from;v=prev_v[v]) { edge &e=E[prev_v[v]][prev_e[v]]; e.capacity -= lc; E[v][e.reve].capacity += lc; } } return res; } }; MinCostFlow<23050,ll> mcf; int N1,N2,M,R,B; string C[2]; vector<pair<int,int>> E[2][202]; pair<int,int> P[202]; void solve() { int i,j,k,l,r,x,y; string s; cin>>N1>>N2>>M>>R>>B; cin>>C[0]>>C[1]; ll diff=0; int in=0,out=0; FOR(i,N1) { if(C[0][i]=='R') { mcf.add_edge(0,1+i,1,-(1<<30)), diff++,in++; mcf.add_edge(0,1+i,500,0); } else if(C[0][i]=='B') { mcf.add_edge(1+i,1000,1,-(1<<30)), diff++,out++; mcf.add_edge(1+i,1000,500,0); } else { mcf.add_edge(0,1+i,500,0); mcf.add_edge(1+i,1000,500,0); } } FOR(i,N2) { if(C[1][i]=='R') { mcf.add_edge(300+i,1000,1,-(1<<30)), diff++,out++; mcf.add_edge(300+i,1000,500,0); } else if(C[1][i]=='B') { mcf.add_edge(0,300+i,1,-(1<<30)), diff++,in++; mcf.add_edge(0,300+i,1000,0); } else { mcf.add_edge(0,300+i,500,0); mcf.add_edge(300+i,1000,500,0); } } FOR(i,M) { cin>>P[i].first>>P[i].second; P[i].first--; P[i].second--; mcf.add_edge(1+P[i].first,300+P[i].second,1,R,i); mcf.add_edge(300+P[i].second,1+P[i].first,1,B,i); } ll ret=diff<<30; while(1) { ll pat=mcf.mincost(0,1000,1); if(pat>0) break; ret+=pat; } if(ret>1LL<<29) { cout<<-1<<endl; } else { cout<<ret<<endl; string S(M,'U'); FOR(i,N1) FORR(e,mcf.E[1+i]) if(e.id>=0 && e.capacity==0) S[e.id]='R'; FOR(i,N2) FORR(e,mcf.E[300+i]) if(e.id>=0 && e.capacity==0) S[e.id]='B'; cout<<S<<endl; } }
まとめ
1以上のフローが流れることを強制するテクとして面白いな。