zoj How Many Shortest Path

How Many Shortest Path

题目：

给出一张图，求解最短路有几条。处理特别BT。还有就是要特别处理map[i][i] = 0，数据有不等于0的情况！

竟然脑残到了些错floyd！

！。！！

！

14次wrong。！

！。。！

。。

算法：

先最短路处理，把在最短路上的边加入上。既是。dist[s][i] + map[i][j] == dist[s][j]表示从起点到i点加上当前边是最短路。把这个加入到网络流边集中。容量为1.然后，建立一个超级源点。容量为INF。

#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
#include <cstdio>
#include <cstring>
using namespace std;

const int INF = 1 << 30;
const int MAXN = 200 + 10;
struct Edge{
   int from,to,cap,flow;
   Edge(){};
   Edge(int _from,int _to,int _cap,int _flow)
      :from(_from),to(_to),cap(_cap),flow(_flow){};
};
vector<Edge> edges;
vector<int> G[MAXN];
int cur[MAXN],d[MAXN];
bool vst[MAXN];
int N,M,src,sink;

int mat[MAXN][MAXN],dist[MAXN][MAXN];
int ss,tt;

void init(){
    for(int i = 0;i <= N+5;++i)
        G[i].clear();
    edges.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));
   int sz = edges.size();
   G[from].push_back(sz - 2);
   G[to].push_back(sz - 1);
}

void flody(){
    memcpy(dist,mat,sizeof(mat));

    for(int i = 0;i < N;++i)
        for(int j = 0;j < N;++j)
           dist[i][j] = (dist[i][j]==-1?

INF:dist[i][j]);

    for(int k = 0;k < N;++k)
        for(int i = 0;i < N;++i)
           for(int j = 0;j < N;++j){
//             if(dist[i][k] == INF||dist[k][j] == INF) continue;
             if(dist[i][k] < INF && dist[k][j] < INF&&dist[i][j] > dist[i][k] + dist[k][j])
                dist[i][j] = dist[i][k] + dist[k][j];
           }
}

bool BFS(){
    memset(vst,0,sizeof(vst));
    queue<int> Q;
    Q.push(src);
    d[src] = 0;
    vst[src] = 1;

    while(!Q.empty()){
        int u = Q.front(); Q.pop();
        for(int i = 0;i < (int)G[u].size();++i){
            Edge& e = edges[G[u][i]];
            if(!vst[e.to] && e.cap > e.flow){
                vst[e.to] = 1;
                d[e.to] = d[u] + 1;
                Q.push(e.to);
            }
        }
    }

    return vst[sink];
}

int DFS(int x,int a){
   if(x == sink || a == 0)
     return a;

   int flow = 0,f;
   for(int& i = cur[x];i < (int)G[x].size();++i){
       Edge& e = edges[G[x][i]];
       if(d[e.to] == d[x] + 1 && (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;
}

int maxFlow(){
   int flow = 0;
   while(BFS()){
       memset(cur,0,sizeof(cur));
       flow += DFS(src,INF);
   }
   return flow;
}

int main()
{
//   freopen("Input.txt","r",stdin);

    while(~scanf("%d",&N)){
         init();
         for(int i = 0;i < N;++i)
            for(int j = 0;j < N;++j){
               scanf("%d",&mat[i][j]);
            }

         for(int i = 0;i < N;++i)
           mat[i][i] = 0;

         scanf("%d%d",&ss,&tt);
         flody();

         if(ss == tt){
            printf("inf\n");
            continue;
         }

         src = N + 2; sink = tt;
         addEdge(src,ss,INF);
         // 建图
         for(int i = 0;i < N;++i){
            if(dist[ss][i] == INF) continue;
            for(int j = 0;j < N;++j){
                if(i == j) continue;
                if(dist[ss][j] == INF||mat[i][j] == -1) continue;
                if(dist[ss][i] + mat[i][j] == dist[ss][j]){   //该边是否在最短路上
                   addEdge(i,j,1);
                }
            }
         }
         printf("%d\n",maxFlow());
    }
    return 0;
}