ZOJ 2760 How Many Shortest Path（floyd + 最大流）

思路：

首先用floyd跑出最短路，然后对于每条边判断，dis[j]==dis[i]+<i,j>

满足条件即加到网络中，（i，j，1），求解最大流，即为答案。

坑点：

注意他给的样例有诈。。

当i等于j的时候需要自己手动置0，而不是指望他给你0。。

靠。因为这个狂wa不止。。

#include <iostream>
#include <cstdio>
#include <queue>
#include <string.h>
#define inf 0x3f3f3f3f
typedef long long int lli;
using namespace std;

const int maxn = 5200;
struct edge{
lli to,next,v;
}ed[600000+20000];
lli ans;
lli head[maxn],dis[maxn],cnte,cur[maxn];
void ini(){
memset(head,-1,sizeof(head));
ans = 0;cnte = 0;
}
inline void ae(int x,int y,int v){
ed[cnte].to = y;
ed[cnte].v = v;
ed[cnte].next=head[x];
head[x]=cnte++;

ed[cnte].to = x;
ed[cnte].v = 0;
ed[cnte].next=head[y];
head[y]=cnte++;
}
int bfs(int s,int t){
memset(dis,-1,sizeof(dis));
queue<int> q;q.push(s);
dis[s] = 0;
int now;
while(!q.empty()){
now = q.front();q.pop();
for(int i = head[now];i!=-1;i=ed[i].next){
int v = ed[i].to;
if(dis[v] == -1 && ed[i].v > 0){
dis[v] = dis[now]+1;q.push(v);
if(v == t) return 1;
}
}
}
if(dis[t] < 0) return 0;
return 1;
}
lli dfs(int s,int t,lli low){
if(s==t || !low) return low;
lli a,ret = 0;
for(lli &i = cur[s];i!=-1;i=ed[i].next){
int v= ed[i].to;
if(dis[v]==dis[s]+1 && ed[i].v>0 && (a=dfs(v,t,min(low,ed[i].v)))){
ed[i].v -= a; ed[i^1].v += a;
low -= a; ret += a;
if(!low) break;
}
}
return ret;
}

void dinic(int s,int t){
while(bfs(s,t)){
memcpy(cur,head,sizeof(head));
ans += dfs(s,t,inf);
}
}

int ma[110][110];
int ma2[110][110];

void floyd(int n){
for(int k = 1;k <= n;k++){
for(int i = 1;i <= n;i++){
for(int j = 1;j <= n;j++){
ma2[i][j] = min(ma2[i][j],ma2[i][k]+ma2[k][j]);
}
}
}
}
int ddis[maxn];

int main(){
int cas,n;
while(~scanf("%d",&n)){
int temp,s,t;
ini();
memset(ma,0x3f,sizeof(ma));
for(int i = 1;i <= n;i++){
for(int j = 1;j <= n;j++){
scanf("%d",&temp);
if(temp != -1)
ma[i][j] = temp;
if(i == j)
ma[i][j] = 0;
}
}
scanf("%d%d",&s,&t);
if(s==t){
puts("inf");continue;
}
memcpy(ma2,ma,sizeof(ma));
floyd(n);
for(int i = 1;i <= n;i++){
ddis[i] = ma2[s+1][i];
}
for(int i = 1;i <= n;i++){
if(ddis[i] == inf) continue;
for(int j = 1;j <= n;j++){
if(ma[i][j] != inf &&i!=j&& ddis[j] == ddis[i]+ma[i][j]){
ae(i,j,1);
}
}
}
dinic(s+1,t+1);
printf("%lld\n",ans);
}

return 0;
}