3416 Marriage Match IV

[code]//题目意思：给出一个有向图，给出起点和终点，一个人从起点到终点去约会，
//每次走的路程要求是最小。没条路不能重复走，问可以走多少次。

//思路：如果没限制直接就可以跑最大流了，
//加了限制后就是说只有最短路径的边才能走，先求出起点到所有点的最短距离，
//然后找出每条边的起点的距离加上边的权值=终点的距离，那么这条边就可以走
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define inf -0x3f3f3f3f
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define mem(a, b) memset(a, b, sizeof(a))
typedef long long ll;
const int maxn=1010;
struct Edge{
    int from,to,flow,cap;
    Edge(int u,int v,int w,int f):
        from(u),to(v),cap(w),flow(f){}
};

struct Dinic{
    int n,m,s,t;                        //结点数,边数（包括反相弧），源点编号和汇点编号
    vector<Edge>edges;                    //边表，edges[e]和edges[e^1]互为反相弧
    vector<int>G[maxn];                    //邻接表，G[i][j]表示结点i的第j条边在e数组中的序号
    bool vis[maxn];                        //BFS使用
    int d[maxn];                        //从起点到i的距离
    int cur[maxn];                        //当前弧下标
    void AddEdge(int from,int to,int cap){
        edges.push_back(Edge(from,to,cap,0));
        edges.push_back(Edge(to,from,0,0));
        m=edges.size();
        G[from].push_back(m-2);
        G[to].push_back(m-1);
    }

    void init(int n){
        this->n=n;
        for(int i=0;i<=n;i++)
            G[i].clear();
        edges.clear();
    }

    bool BFS(){
        mem0(vis);
        queue<int>Q;
        Q.push(s);
        d[s]=0;
        vis[s]=1;
        while(!Q.empty()){
            int x=Q.front();
            Q.pop();
            for(int i=0;i<G[x].size();i++){                    //只考虑残量网络中的弧
                Edge& e=edges[G[x][i]];
                if(!vis[e.to]&&e.cap>e.flow){
                    vis[e.to]=1;
                    d[e.to]=d[x]+1;
                    Q.push(e.to);
                }
            }
        }
        return vis[t];
    }

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

    ll Maxflow(int s,int t){
        this->s=s;
        this->t=t;
        ll flow=0;
        while(BFS()){
            mem0(cur);
            flow+=DFS(s,INF);
        }
        return flow;
    }
};
struct Edge1{
    int from,to,len;
    Edge1(int u,int v,int c):
        from(u),to(v),len(c){}
};

vector<Edge1>edges;
vector<int>G[maxn];
int dist[maxn];
int vis[maxn];
void init(int n){
    for(int i=0;i<=n;i++)
        G[i].clear();
    edges.clear();
}

void Addedge(int u,int v,int c){
    edges.push_back(Edge1(u,v,c));
    int m=edges.size();
    G[u].push_back(m-1);
}

void spfa(int s){
    memset(dist,0x3f,sizeof(dist));
    mem0(vis);
    queue<int>Q;
    Q.push(s);
    dist[s]=0;
    vis[s]=1;
    while(!Q.empty()){
        int x=Q.front();
        Q.pop();
        vis[x]=0;                         //该开始忘记了松弛操作 
        for(int i=0;i<G[x].size();i++){
            Edge1& e=edges[G[x][i]];
            if(dist[e.to]>dist[x]+e.len){
                dist[e.to]=dist[x]+e.len;
                if(!vis[e.to]){
                    vis[e.to]=1;
                    Q.push(e.to);
                }
            }
        }
    }
}

Dinic ZYC;
int main(){
    int t;
    int n,m;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        init(n);
        int x,y,z;
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&x,&y,&z);
            if(x!=y)
                Addedge(x,y,z);
        }
        int start,end;
        scanf("%d%d",&start,&end);
        spfa(start);
        ZYC.init(n);
        for(int i=1;i<=n;i++){
            for(int j=0;j<G[i].size();j++){
                Edge1& e=edges[G[i][j]];
                if(dist[e.to]==dist[i]+e.len){
                    ZYC.AddEdge(i,e.to,1);
                }
            }
        }
        printf("%I64d\n",ZYC.Maxflow(start,end));
    }
    return 0;
}