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

题目大意：在城市A的男孩想去城市B的女孩，不过他去城市B必须走最短路，并且走过的路不可以再走，问他最多能看这个女孩多少次。

 

分析：因为这个男孩直走最短路，所以我们必须求出来所有最短路径上的路，怎么判断一条路是否属于最短路经上的呢？其实比较容易的，只要先求出来从A到达所有点的最短路distA[x], 然后再求出来所有点到B的最短路distB[y]（添加反边从B开始即可求出）,如果x-y之间有一条路，那么只需要判断distA[x]+distB[y]+w(x,y） == distA[B] 是否成立即可。然后把这条边加入新图中，流量置为1（因为只可以走1次），求出来从源点到汇点的最大流就是所求结果。

注意：可能会爆栈。

下面是AC代码。

========================================================================================================================

#pragma comment(linker, "/STACK:102400000,102400000")
#include<stdio.h>
#include<string.h>
#include<queue>
#include<stack>
#include<vector>
using namespace std;

const int MAXN = 1e6+;
const int MAXM = ;
const int oo = 1e9+;
///分别存正向图，反向图，和重构图
struct Edge{int v, val, next;}ss[MAXN], ee[MAXN], edge[MAXN<<];
int Hs[MAXM], He[MAXM], Head[MAXM], cnt_s, cnt_e, cnt;
int DistS[MAXM], DistE[MAXM];///分别是从源点到达每点的距离，和每点到达源点的距离
int Layer[MAXM];///分层

void InIt(int N)
{
    cnt = cnt_e = cnt_s = ;

    memset(Hs, -, sizeof(Hs));
    memset(He, -, sizeof(He));
    memset(Head, -, sizeof(Head));

    for(int i=; i<=N; i++)
        DistE[i] = DistS[i] = oo;
}
void AddEdge(Edge e[], int head[], int &ct, int u, int v, int val)
{
    e[ct].v = v;
    e[ct].val = val;
    e[ct].next = head[u];
    head[u] = ct++;
}
void spfa(Edge e[], int head[], int dist[], int start)
{///求最短路
    stack<int> sta;
    bool instack[MAXM] = {};
    dist[start] = ;
    sta.push(start);

    while(sta.size())
    {
        int u = sta.top();sta.pop();
        instack[u] = false;

        for(int j=head[u]; j!=-; j=e[j].next)
        {
            int v = e[j].v;

            if(dist[v] > dist[u]+e[j].val)
            {
                dist[v] = dist[u] + e[j].val;

                if(instack[v] == false)
                {
                    instack[v] = true;
                    sta.push(v);
                }
            }
        }
    }
}
void BuildGraph(int u, int MinLen)
{///遍历所有的边，把输入最短路的边加入新图中
    for(int j=Hs[u]; j!=-; j=ss[j].next)
    {
        int v = ss[j].v;

        if(v != -)
        {
            if(DistS[u] + DistE[v] + ss[j].val == MinLen)
            {
                AddEdge(edge, Head, cnt, u, v, );
                AddEdge(edge, Head, cnt, v, u, );
            }

            ss[j].v = -;

            BuildGraph(v, MinLen);
        }
    }
}
bool BFS(int start, int End)
{
    memset(Layer, , sizeof(Layer));
    Layer[start] = ;
    queue<int> Q;
    Q.push(start);

    while(Q.size())
    {
        int u = Q.front();Q.pop();

        if(u == End)return true;

        for(int j=Head[u]; j!=-; j=edge[j].next)
        {
            int v = edge[j].v;

            if(Layer[v] == false && edge[j].val)
            {
                Layer[v] = Layer[u] + ;
                Q.push(v);
            }
        }
    }

    return false;
}
int DFS(int u, int MaxFlow, int End)
{
    if(u == End)return MaxFlow;

    int uflow = ;

    for(int j=Head[u]; j!=-; j=edge[j].next)
    {
        int v = edge[j].v;

        if(Layer[u]+ == Layer[v] && edge[j].val)
        {
            int flow = min(MaxFlow-uflow, edge[j].val);
            flow = DFS(v, flow, End);

            edge[j].val -= flow;
            edge[j^].val += flow;
            uflow += flow;

            if(uflow == MaxFlow)
                break;
        }
    }

    if(uflow == )
        Layer[u] = ;

    return uflow;
}
int Dinic(int start, int End)
{
    int MaxFlow = ;

    while(BFS(start, End) == true)
        MaxFlow += DFS(start, oo, End);

    return MaxFlow;
}

int main()
{
    int T;

    scanf("%d", &T);

    while(T--)
    {
        int N, M, u, v, val;

        scanf("%d%d", &N, &M);
        InIt(N);

        while(M--)
        {
            scanf("%d%d%d", &u, &v, &val);

            if(u == v)continue;

            AddEdge(ss, Hs, cnt_s, u, v, val);
            AddEdge(ee, He, cnt_e, v, u, val);
        }

        int start, End;

        scanf("%d%d", &start, &End);

        spfa(ss, Hs, DistS, start);///求源点到所有点的距离
        spfa(ee, He, DistE, End);///求所有点到汇点的最短距离
        BuildGraph(start, DistS[End]);///构建新图

        printf("%d\n", Dinic(start, End));
    }

    return ;
}