ZOJ 1760 How Many Shortest Path (网络流.水)

题意是， 给你一个有向带权图，给你源点汇点S, T， 你会得到S->T 的最短路，求有多少条不同的最短路（不同：没有公共边）
做法很简单，先求一遍最短路（spfa/fload) 之后建立新图， 如果dis[i] + edge[i][j] == dis[j] 就添加边edge[i][j]。流量为1
之后S->T求一次最大流， 流量就是结果。

不知道第一次哪儿敲错了tle, 第二次才过。T_T

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<cmath>
#include<queue>
#include<map>
#include<set>
using namespace std;
#define INF 1000000000
//typedef __int64 LL;
#define N 111
int n, mat[N][N], tot, hh[N], dis[N], inq[N], lev[N], S, T;
struct node
{
    int u, v, w, next;
}edge[1000000];
void init()
{
    memset(hh, -1, sizeof(hh));
    tot  = 0;
}
void add(int u, int v,int w)
{
    edge[tot].u = u; edge[tot].v = v;
    edge[tot].w = w; edge[tot].next = hh[u];
    hh[u] = tot ++;
}
void spfa()
{
    queue<int > Q;
    for(int i=0; i<=n; i++) dis[i] = INF;
    dis[S] = 0; inq[S] = 1; Q.push(S);
    while(!Q.empty()) {
        int u = Q.front(); Q.pop();
        inq[u] = 0;
        for(int v= 1; v<=n; v++) {
            if(mat[u][v]<0 ) continue; // ?
            if(dis[v] > dis[u] + mat[u][v]) {
                dis[v] = dis[u] + mat[u][v];
                if(!inq[v]) {
                    inq[v] = 1; Q.push(v);
                }
            }
        }
    }
}
int bfs()
{
    queue<int > Q;
    memset(lev, -1, sizeof(lev));
    Q.push(S); lev[S] = 0;
    while(!Q.empty()) {
        int u = Q.front(); Q.pop();
        for(int i=hh[u]; i!=-1; i=edge[i].next) {
            int v = edge[i].v;
            if(edge[i].w && lev[v]==-1) {
                lev[v] = lev[u] +1;
                Q.push(v);
            }
        }
    }
    return lev[T] != -1;
}
int dfs(int u, int flow)
{
    if(u== T) return flow;
    int tmp = flow, ad;
    for(int i=hh[u]; i!=-1; i=edge[i].next) {
        int v = edge[i].v;
        int w = edge[i].w;
        if(lev[v] == lev[u] +1 && tmp>0 && edge[i].w) {
            ad = dfs(v, min(tmp, edge[i].w));
            if(!tmp) break;
            edge[i].w -= ad;
            edge[i^1].w += ad;
            tmp -= ad;
        }
    }
    if(ad == 0) lev[u] = -1;
    return flow - tmp;
}
int dinic()
{
    int ret = 0, tmp;
    while(bfs()) {
        while(tmp = dfs(S, INF)) ret += tmp;
    }
    return ret ;
}
int main()
{
#ifndef ONLINE_JUDGE
	freopen("in.txt", "r", stdin);
#endif // ONLINE_JUDGE
	while(scanf("%d", &n)!=EOF) {
        for(int i=1; i<=n; i++) {
            for(int j=1; j<=n; j++) {
                scanf("%d", &mat[i][j]);
            }
        }
        scanf("%d%d", &S, &T);
        S++; T++;
        if(S == T)
        {
            printf("inf\n"); continue;
        }
        spfa();
        init();
        for(int i=1; i<=n; i++) {
            for(int j=1; j<=n; j++) {
                if(i==j || mat[i][j]<0) continue;
                if(dis[i] + mat[i][j] == dis[j]) {
                    add(i, j,1); add(j, i, 0);
                }
            }
        }
        int ans = dinic();
        printf("%d\n", ans);
	}
	return 0;
}