1. 程式人生 > >poj3680 Intervals 區間k覆蓋問題 最小費用最大流 建圖巧妙

poj3680 Intervals 區間k覆蓋問題 最小費用最大流 建圖巧妙

全部 cstring ras 如果 cos printf als lld map

/**
題目:poj3680 Intervals 區間k覆蓋問題 最小費用最大流 建圖巧妙
鏈接:http://poj.org/problem?id=3680
題意:給定n個區間,每個區間(ai,bi),以及權值wi。選出一些區間,滿足權值和最大且任何一個點不會被超過k個區間覆蓋。
思路:
建圖:對於每個區間(ai,bi)。 ai->bi,cap = 1,cost = -wi; (離散化後的ai,bi)
所有區間的端點放到數組,進行從小到大排序,去重,離散化,在數組內相鄰的u端點,v端點。u->v,cap = INF,cost=0;
s->x最左邊的那個端點(也就是離散化後最小的那個數),cap = k, cost = 0;
x(最右邊的那個端點)->t,cap = k, cost = 0;
求s->t的最小費用最大流,輸出-cost即為結果。

可以把圖想象成一個x軸上有若幹端點,(ai,bi)上面連了一條弧線。從最左邊開始跑,到最右邊,如果最左邊cap=k。
那麽最多k流量往右邊流,每個點最多被k流量覆蓋,每一個單位流量分配給一個區間(ai,bi),所以最多被k個區間覆蓋。

*/ #include<iostream> #include<cstring> #include<vector> #include<map> #include<cstdio> #include<sstream> #include<algorithm> #include<queue> using namespace std; typedef long long LL; const int INF = 0x3f3f3f3f; const int N = 420; struct Edge{ int from
, to, cap, flow, cost; Edge(int u,int v,int c,int f,int w):from(u),to(v),cap(c),flow(f),cost(w){} }; struct MCMF{ int n, m; vector<Edge> edges; vector<int> G[N]; int inq[N]; int d[N]; int p[N]; int a[N]; void init(int n){ this->n = n;
for(int i = 0; i <= n; i++) G[i].clear(); edges.clear(); } void AddEdge(int from,int to,int cap,long long cost){ edges.push_back(Edge(from,to,cap,0,cost)); edges.push_back(Edge(to,from,0,0,-cost)); m = edges.size(); G[from].push_back(m-2); G[to].push_back(m-1); } bool BellmanFord(int s,int t,int &flow,long long &cost){ for(int i = 0; i <= n; i++) d[i] = INF; memset(inq, 0, sizeof inq); d[s] = 0; inq[s] = 1; p[s] = 0; a[s] = INF; queue<int> Q; Q.push(s); while(!Q.empty()){ int u = Q.front(); Q.pop(); inq[u] = 0; for(int i = 0; i < G[u].size(); i++){ Edge& e = edges[G[u][i]]; if(e.cap>e.flow&&d[e.to]>d[u]+e.cost){ d[e.to] = d[u]+e.cost; p[e.to] = G[u][i]; a[e.to] = min(a[u],e.cap-e.flow); if(!inq[e.to]) {Q.push(e.to); inq[e.to] = 1;} } } } if(d[t]==INF) return false; flow += a[t]; cost += (long long)d[t]*(long long)a[t]; for(int u = t; u!=s; u = edges[p[u]].from){ edges[p[u]].flow+=a[t]; edges[p[u]^1].flow-=a[t]; } return true; } int MincostMaxflow(int s,int t,long long &cost){ int flow = 0; cost = 0; while(BellmanFord(s,t,flow,cost)); return flow; } }; int u[N], v[N], w[N]; vector<int> vv; map<int,int> mp; int main() { int T, n, k; cin>>T; while(T--) { int s = 0, t; scanf("%d%d",&n,&k); vv.clear(); for(int i = 1; i <= n; i++){ scanf("%d%d%d",&u[i],&v[i],&w[i]); vv.push_back(u[i]); vv.push_back(v[i]); } sort(vv.begin(),vv.end()); vv.erase(unique(vv.begin(),vv.end()),vv.end());///相鄰重復元素,多余出來的全部放到後面,並返回一個開始指針 mp.clear(); for(int i = 0; i < vv.size(); i++){ mp[vv[i]] = i+1; } t = vv.size()+1; MCMF mcmf; mcmf.init(t); mcmf.AddEdge(s,1,k,0); mcmf.AddEdge(vv.size(),t,k,0); for(int i = 1; i < vv.size(); i++){ mcmf.AddEdge(i,i+1,INF,0); } for(int i = 1; i <= n; i++){ mcmf.AddEdge(mp[u[i]],mp[v[i]],1,-w[i]); } LL cost; int flow = mcmf.MincostMaxflow(s,t,cost); printf("%lld\n",-cost); } return 0; }

poj3680 Intervals 區間k覆蓋問題 最小費用最大流 建圖巧妙