HDU-3416 Marriage Match IV （最短路+最大流）

Marriage Match IV

题目链接：Marriage Match IV HDU - 3416

题意

给你一个图，问有多少条不同的最短路，有重边并且可当作分开的两边，其多条最短路之间不能有任何相同的边

---

思路

开始时，看到了每条边只能经过一次，我想到了费用流，可惜这里的数据不小，N为1000，M为1e5，按照费用流平均复杂度，F * E * log（v） 毫无疑问会T，那么可以怎么优化呢。

我们可以先跑一下最短路，找到找到所有可能组成最短路的边，这个判断就是 dis[to] == dis[now] + cost，如果满足，则一定是最短路中的关键组成。

既然有了这些边，那么我们再以这些边组成一个网络，跑一个最大流即可。

---

代码

#include <bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = (int)j;i <= (int)k;i ++)
#define per(i,j,k) for(int i = (int)j;i >= (int)k;i --)
#define debug(x) cerr<<#x<<" = "<<(x)<<endl
#define mmm(a,b) memset(a,b,sizeof(a))
#define pb push_back

typedef double db;
typedef long long ll;
const int MAXN = (int)1e3+7;
const int INF = (int)0x3f3f3f3f;

// 用于表示表示边的结构体（终点、容量、反向边）
struct edge{
    int to,cap,rev;
    edge(int to = 0,int cap = 0,int rev = 0):to(to),cap(cap),rev(rev){}
};

int N,M;
vector<edge> G[MAXN];
int level[MAXN]; //顶点到源点的距离标号
int iter[MAXN];  //当前弧，在其之前的边已经没有用了

//向图中增加一条从s到t容量为cap的边
void add_edge(int from,int to,int cap) {
    G[from].pb(edge(to,cap,G[to].size()));
    G[to  ].pb(edge(from,0,G[from].size()-1));
}

//通过BFS计算从源点出发的距离标号
void bfs(int s){
    mmm(level,-1);
    queue<int> qu;
    level[s] = 0;
    qu.push(s);
    while (!qu.empty()) {
        int v = qu.front(); qu.pop();
        rep(i,0,G[v].size()-1) {
            edge &e = G[v][i];
            if (e.cap > 0 && level[e.to] < 0) {
                level[e.to] = level[v] + 1;
                qu.push(e.to);
            }
        }
    }
}

//通过DFS寻找增广路
int dfs(int v,int t,int f) {
    if (v == t) return f;
    for (int &i = iter[v];i < G[v].size();i ++) {
        edge &e = G[v][i];
        if (e.cap > 0 && level[v] < level[e.to]) {
            int d = dfs(e.to,t,min(f,e.cap));
            if (d > 0) {
                e.cap -= d;
                G[e.to][e.rev].cap += d;
                return d;
            }
        }
    }
    return 0;
}

//求解从s到t的最大流
int max_flow(int s,int t){
    int flow = 0;
    for (;;) {
        bfs(s);
        if (level[t] < 0) return flow;
        mmm(iter,0);
        int f;
        while ((f = dfs(s,t,INF)) > 0)
            flow += f;
    }
}

int pic[MAXN][MAXN];
int dis[MAXN];


struct Node{
    int to,cost;
    Node(int to = 0,int cost = 0):to(to),cost(cost){}
    bool operator < (const Node&a) const {
        return a.cost < cost;
    }
};
priority_queue<Node> qu;
vector<Node> E[MAXN];
void init(){
    rep(i,1,N) E[i].clear();
    rep(i,1,N) G[i].clear();
    fill(dis+1,dis+1+N,INF);
    mmm(pic,0);
}

void Djs(int s,int t) {
    while (!qu.empty()) qu.pop();
    qu.push(Node(s,0));
    dis[s] = 0;
    while (!qu.empty()) {
        Node k = qu.top();qu.pop();
        rep(i,0,E[k.to].size()-1) {
            int to = E[k.to][i].to;
            int cost = E[k.to][i].cost;
            if (dis[to] > dis[k.to] + cost) {
                dis[to] = dis[k.to] + cost;
                qu.push(Node(to,dis[to]));
            }
        }
    }
}

int main()
{
    int T;
    scanf("%d",&T);
    while (T --) {
        scanf("%d %d",&N,&M);
        init();
        rep(i,1,M) {
            int u,v,c;
            scanf("%d %d %d",&u,&v,&c);
            if (u == v) continue;
            E[u].pb(Node(v,c));
        }
        int s,t;scanf("%d %d",&s,&t);
        Djs(s,t);
        rep(i,1,N) {
            rep(j,0,E[i].size()-1) {
                int to = E[i][j].to;
                int cost = E[i][j].cost;
                if (dis[to] == dis[i] + cost) pic[i][to] ++;
            }
        }
        rep(i,1,N) {
            rep(j,1,N) {
                if (pic[i][j]) add_edge(i,j,pic[i][j]);
            }
        }
        printf("%d\n",max_flow(s,t));
    }
}