首页 > 代码库 > zoj2676 Network Wars(0-1分数规划,最大流模板)
zoj2676 Network Wars(0-1分数规划,最大流模板)
Network Wars
07年胡伯涛的论文上的题:http://wenku.baidu.com/view/87ecda38376baf1ffc4fad25.html
代码:
#include <algorithm> #include <cstdio> #include <iterator> #include <limits> #include <vector> #include <string.h> const int N = 111; const int M = 404; const double EPS = 1e-3; typedef std::vector<int> VI; int signum(double x) { return x > EPS ? 1 : (x < -EPS ? -1 : 0); } template<int N, int M, class Flow> struct Dinic { int n, e, first[N], cur[N], next[M], to[M], id[M], s, t; int pre[N], level[N], q[N], sign; Flow cap[M], flow; void add(int u, int v, Flow w, int i) { to[e] = v; cap[e] = w; id[e] = i; next[e] = first[u]; first[u] = e++; } bool bfs(int s, int t) { std::fill(level+1, level + n + 1, -1); sign = t; level[t] = 0; int head = 0, tail = 0; q[tail++] = t; while (head != tail && level[s] == -1) { int u = q[head++]; for (int it = first[u]; it != -1; it = next[it]) { if (cap[it ^ 1] > 0 && level[to[it]] == -1) { level[to[it]] = level[u] + 1; q[tail++] = to[it]; } } } return level[s] != -1; } void push() { Flow delta = std::numeric_limits<Flow>::max(); int u, p; for (u = t; u != s; u = to[p ^ 1]) { p = pre[u]; delta = std::min(delta, cap[p]); } for (u = t; u != s; u = to[p ^ 1]) { p = pre[u]; cap[p] -= delta; if (!cap[p]) { sign = to[p ^ 1]; } cap[p ^ 1] += delta; } flow += delta; } void dfs(int u) { if (u == t) { push(); } else { for (int & it = cur[u]; it != -1; it = next[it]) { if (cap[it] > 0 && level[u] == level[to[it]] + 1) { pre[to[it]] = it; dfs(to[it]); if (level[sign] > level[u]) { return; } sign = t; } } level[u] = -1; } } void init(int _n, int _s, int _t) { n = _n, s = _s, t = _t; std::fill(first + 1 , first + n + 1 , -1); e = 0; } Flow solve() { flow = 0; while (bfs(s, t)) { for (int i = 1; i <= n; ++i) { cur[i] = first[i]; } dfs(s); } return flow; } }; Dinic<N,M<<1,double> AC ; int left[M] , right[M] , w[M] ; int n , m , mark[M] ; double judge ( double key ) { double sum = 0 ; memset ( mark , 0 , sizeof ( mark ) ) ; AC.init ( n , 1 , n ) ; for ( int i = 1 ; i <= m ; i ++ ) if ( w[i] <= key ) { sum += w[i] - key ; mark[i] = 1 ; } else { AC.add ( left[i] , right[i] , w[i] - key , i ) ; AC.add ( right[i] , left[i] , w[i] - key , i ) ; } double add = AC.solve () ; sum += add ; return sum ; } void solve () { double l = 0 , r = (double) m * 1e7 ; while ( signum (r-l) > 0 ) { double mid = ( l + r ) / 2.0 ; double k = judge ( mid ) ; if ( signum ( k ) >= 0 ) l = mid ; else r = mid ; } AC.bfs ( 1 , n ) ; for ( int i = 0 ; i < AC.e ; i ++ ) { int u = AC.to[i^1] , v = AC.to[i] ; if ( AC.level[u] == -1 && AC.level[v] != -1 ) mark[AC.id[i]] = 1 ; } int ans = 0 ; for ( int i = 1 ; i <= m ; i ++ ) if ( mark[i] ) ans ++ ; printf ( "%d\n" , ans ) ; for ( int i = 1 ; i <= m ; i ++ ) if ( mark[i] ) printf ( "%d " , i ) ; puts ( "" ) ; } int main () { // freopen("network.in", "r", stdin); // freopen("network.out", "w", stdout); int flag = 0 ; while ( scanf ( "%d%d" , &n , &m ) != EOF ) { for ( int i = 1 ; i <= m ; i ++ ) scanf ( "%d%d%d" , &left[i] , &right[i] , &w[i] ) ; if ( flag ) puts ( "" ) ; flag = 1 ; solve () ; } return 0 ; }
zoj2676 Network Wars(0-1分数规划,最大流模板)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。