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

, tot, hh
, dis
, inq
, lev
, 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;
}