POJ 1895 Bring Them There 分层构图 求最大流 并输出路径

这个题在输出路径上上纠结了好久,直到我遇到了下面这个代码,,堪称完美!

题目限时5s,只用了1s;


Code:

    //神一样的思想,神一样的代码,我佩服的五体投地!!  
    //题目给出五个正整数n,m,k,s,t。n表示n星球,m表示m个双向隧道,  
    //k,s,t表示要把k个电脑从s星球运到t星球,双向道只能单行,走一条  
    //道用时一天,求最短时间,并输出运输方案。  
    //分析:  
    //假定T天可以运完,可以吧每个点u拆分成T+1个,分别为u0,u1。。。  
    //u0是初始状态,u1是一天后的状态,对于相邻的a,b。a0到b1有一条  
    //容量为1的边,b0到a1也有一条,a0到a1有一天容量为INF的路,表示不动。  
    //用模板求出最大流,如果达到了k就输出T,达不到就T++;  
    #include<cstdio>  
    #include<cstring>  
    #include<queue>  
    #include<vector>  
    #include<algorithm>  
    using namespace std;  
      
    const int maxn = 5000 + 10;  
    const int INF = 1000000000;  
      
    struct Edge {  
      int from, to, cap, flow;  
    };  
      
    bool operator < (const Edge& a, const Edge& b) {  
      return a.from < b.from || (a.from == b.from && a.to < b.to);  
    }  
      
    struct Dinic {  
      int n, m, s, t;  
      vector<Edge> edges;    // 边数的两倍  
      vector<int> G[maxn];   // 邻接表,G[i][j]表示结点i的第j条边在e数组中的序号  
      bool vis[maxn];        // BFS使用  
      int d[maxn];           // 从起点到i的距离  
      int cur[maxn];         // 当前弧指针  
      
      void init() { edges.clear(); }  
      
      void clearNodes(int a, int b) {  
        for(int i = a; i <= b; i++) G[i].clear();  
      }  
      
      void AddEdge(int from, int to, int cap) {  
        edges.push_back((Edge){from, to, cap, 0});  
        edges.push_back((Edge){to, from, 0, 0});  
        m = edges.size();  
        G[from].push_back(m-2);  
        G[to].push_back(m-1);  
      }  
      
      bool BFS() {  
        memset(vis, 0, sizeof(vis));  
        queue<int> Q;  
        Q.push(s);  
        vis[s] = 1;  
        d[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(!vis[e.to] && e.cap > e.flow) {  
              vis[e.to] = 1;  
              d[e.to] = d[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(d[x] + 1 == d[e.to] && (f = DFS(e.to, min(a, e.cap-e.flow))) > 0) {  
            e.flow += f;  
            edges[G[x][i]^1].flow -= f;  
            flow += f;  
            a -= f;  
            if(a == 0) break;  
          }  
        }  
        return flow;  
      }  
      
      // 求s-t最大流。如果最大流超过limit,则只找一个流量为limit的流  
      int Maxflow(int s, int t, int limit) {  
        this->s = s; this->t = t;  
        int flow = 0;  
        while(BFS()) {  
          memset(cur, 0, sizeof(cur));  
          flow += DFS(s, limit - flow);  
          if(flow == limit) break; // 达到流量限制,直接退出  
        }  
        return flow;  
      }  
    };  
      
    Dinic g;  
      
    const int maxm = 200 + 10;  
    int main() {  
      int n, m, k, S, T;  
      int u[maxm], v[maxm]; // 输入边  
      while(scanf("%d%d%d%d%d", &n, &m, &k, &S, &T) == 5) {  
        for(int i = 0; i < m; i++) scanf("%d%d", &u[i], &v[i]);  
        g.init();  
        int day = 1;  
        g.clearNodes(0, n-1); // 第一层结点编号为0~n-1。第day层(day>=1)结点编号为day*n~day*n+n-1  
        int flow = 0;  
        for(;;) {  
          // 判断day天是否有解  
          // 一架飞船最多需要n-1天到达目的地,沿着这一路线最多需要n+k-2天就可以运完所有飞船,总结点数不超过(n+k-1)n  
          g.clearNodes(day*n, day*n+n-1);  
          for(int i = 0; i < n; i++) g.AddEdge((day-1)*n+i, day*n+i, INF); // 原地不动  
          for(int i = 0; i < m; i++) {  
            g.AddEdge((day-1)*n+u[i]-1, day*n+v[i]-1, 1); // u[i]->v[i]  
            g.AddEdge((day-1)*n+v[i]-1, day*n+u[i]-1, 1); // v[i]->u[i]  
          }  
          flow += g.Maxflow(S-1, day*n+T-1, k - flow);//第0天的源点,到第n天的汇点  
          if(flow == k) break;  
          day++;  
        }  
      
        // 输出解  
        printf("%d\n", day);  
        int idx = 0;  
        vector<int> location(k, S); // 每架飞船的当前位置  
        for(int d = 1; d <= day; d++) {  
          idx += n*2;//前2*n个点是自己通向自己的流量(不动时),应该跳过。  
          vector<int> moved(k, 0); // 第d天有没有移动飞船i  
          vector<int> a, b;        // 第d天有一架飞船从a[i]到b[i]  
          for(int i = 0; i < m; i++) {//接下来就是2*m个两个不同的点的流量  
            int f1 = g.edges[idx].flow; idx += 2;  
            int f2 = g.edges[idx].flow; idx += 2;  
            if(f1 == 1 && f2 == 0) { a.push_back(u[i]); b.push_back(v[i]); }  
            if(f1 == 0 && f2 == 1) { a.push_back(v[i]); b.push_back(u[i]); }  
          }  
          printf("%d", a.size());//第d天走的飞船数  
          for(int i = 0; i < a.size(); i++) {  
            // 查找是哪架飞船从a[i]移动到了b[i]  
            for(int j = 0; j < k; j++)//模拟一下整个移动过程  
              if(!moved[j] && location[j] == a[i]) {  
                printf(" %d %d", j+1, b[i]);  
                moved[j] = 1;  
                location[j] = b[i];  
                break;  
              }  
          }  
          printf("\n");  
        }  
      }  
      return 0;  
    }  


©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页