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

题意：

　　求有多少条最短路

解析：

　　正着求一遍最短路 得dis1 反着求一遍得 dis2   然后 遍历所有的边 如果  dis1[u] + dis2[v] + w == dis1[B]， 则说明这是一条最短路边

　　建网络流的边 容量为1 代表这条边只能走一次

　　如果是无向边一次就好了， 就是挑出来所有符合条件的边

　　------------------------------

　　为什么我的代码的规范上边和下边不一样。。。。。因为。。。emm。。。这是我半年前做的题。。然而那时候只会EK 当然T了。。会了Dinic后。。就忘了这题了。。。今天发现 啊啊啊啊啊。。。然后就改了一下代码。。。。我竟然和半年前做一样的题。。。我真是菜啊。。。。。。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <cmath>
#include <vector>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
const int maxn = 1e3 + , INF = 0xfffffff, maxm = ;
typedef long long LL;
int n,m, cnt, s, t;
int head[maxn], d[maxn], vis[maxn], dis1[maxn], dis2[maxn], head1[maxn], cur[maxn];
int from[maxm], to[maxm], w[maxm];

struct edge
{
    int u, v, c, next;
}Edge[maxm << ];

void add_(int u, int v, int c)
{
    Edge[cnt].u = u;
    Edge[cnt].v = v;
    Edge[cnt].c = c;
    Edge[cnt].next = head1[u];
    head1[u] = cnt++;
}

void add_edge(int u, int v, int c)
{
    add_(u, v, c);
    add_(v, u, );
}

bool bfs()
{
    mem(d, );
    queue<int> Q;
    Q.push(s);
    d[s] = ;
    while(!Q.empty())
    {
        int u = Q.front(); Q.pop();
        for(int i = head1[u]; i != -; i = Edge[i].next)
        {
            edge e = Edge[i];
            if(!d[e.v] && e.c > )
            {
                d[e.v] = d[u] + ;
                Q.push(e.v);
                if(e.v == t) return ;
            }
        }
    }
    return d[t] != ;
}

int dfs(int u, int cap)
{
    int ret = ;
    if(u == t || cap == )
        return cap;
    for(int &i = cur[u]; i != -; i = Edge[i].next)
    {
        edge e = Edge[i];
        if(d[e.v] == d[u] +  && e.c > )
        {
            int V = dfs(e.v, min(e.c, cap));
            Edge[i].c -= V;
            Edge[i ^ ].c += V;
            ret += V;
            cap -= V;
            if(cap == ) break;
        }
    }
    if(cap > ) d[u] = -;
    return ret;
}

int Dinic()
{
    int ans = ;
    while(bfs())
    {
        memcpy(cur, head1, sizeof(head1));
        ans += dfs(s, INF);
    }
    return ans;
}

struct node
{
    int u, v, w, next;
}Node[maxn * ];

void add(int u,int v,int w,int i)
{
    Node[i].u = u;
    Node[i].v = v;
    Node[i].w = w;
    Node[i].next = head[u];
    head[u] = i;
}
void spfa(int s)
{
    for(int i = ; i < maxn; i++) d[i] = INF;
    queue<int> Q;
    d[s] = ;
    mem(vis,);
    Q.push(s);
    vis[s] = ;
    while(!Q.empty())
    {
        int u = Q.front();Q.pop();
        vis[u] = ;
        for(int i=head[u]; i!=-; i=Node[i].next)
        {
            node e = Node[i];
            if(d[e.v] > d[u] + e.w)
            {
                d[e.v] = d[u] + e.w;
                if(!vis[e.v])
                {
                    Q.push(e.v);
                    vis[e.v] = ;
                }
            }
        }

    }
}

int main()
{
    int T,A,B;
    scanf("%d",&T);
    while(T--)
    {
        mem(Node,);
        mem(head,-);
        mem(head1, -);
        cnt = ;
        scanf("%d%d",&n,&m);
        for(int i=; i<m; i++)
        {
           scanf("%d%d%d",&from[i],&to[i],&w[i]);
           add(from[i],to[i],w[i],i);
        }
        scanf("%d%d",&A,&B);
        s = A, t = B;
        spfa(A);
        mem(Node,);
        for(int i=; i<=n; i++)
            dis1[i] = d[i];
        mem(head,-);
        for(int i=; i<m; i++)
            add(to[i],from[i],w[i],i);
        spfa(B);
        for(int i=; i<=n; i++)
            dis2[i] = d[i];
        for(int i=; i<m; i++)
            if(from[i] != to[i] && dis1[from[i]] + dis2[to[i]] + w[i] == dis1[B])
                add_edge(from[i],to[i],);

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