2019牛客暑期多校训练营（第一场） - Path 双向最短路+dinic

#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#define make pair<int,int>
using namespace std;
typedef long long ll;
const ll inf =1e16,maxinf=1e9+10;
const int maxn=1e4+10;
int n,m;
struct node{
    ll d;int x;
    bool operator < (const node &a)const{
        return d>a.d;
    }
};
int vis[maxn],uu[maxn],vv[maxn],dd[maxn];
ll dis1[maxn],dis2[maxn];
vector<make>mp[maxn];
priority_queue<node>q;
void dijkstra(int s,ll *dis){
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++)dis[i]=inf;
    dis[s]=0;
    q.push({0,s});
//    for(int i=1;i<=n;i++)
//        cout<<dis[i]<<" ";
//    cout<<endl;
    while(!q.empty()){
//        cout<<"asd"<<endl;
        node t=q.top();q.pop();
        int u=t.x;
        if(vis[u])continue;
        vis[u]=1;
        int len=mp[u].size();
        for(int i=0;i<len;i++){
            int v=mp[u][i].first;
            int w=mp[u][i].second;
            if(dis[v]>dis[u]+w){
                dis[v]=dis[u]+w;
                q.push({dis[v],v});
            }
        }
    }
    return;
}
struct edge{
    int to,cap,rev;
};
vector<edge>G[maxn];
int level[maxn],iter[maxn];
void add(int u,int v,int cap){
    G[u].push_back({v,cap,G[v].size()});
    G[v].push_back({u,0,G[u].size()-1});
    return ;
}
void bfs(int s){
    memset(level,-1,sizeof(level));
    queue<int>que;
    level[s]=0;
    que.push(s);
    while(!que.empty()){
        int v=que.front();
        que.pop();
        for(int i=0;i<G[v].size();i++){
            edge &e=G[v][i];
            if(e.cap>0&&level[e.to]<0){
                level[e.to]=level[v]+1;
                que.push(e.to);
            }
        }
    }
    return;
}
int dfs(int v,int t,int f){
    if(v==t)return f;
    for(int &i=iter[v];i<G[v].size();i++){
        edge &e=G[v][i];
        if(e.cap>0&&level[v]<level[e.to]){
            int d=dfs(e.to,t,min(f,e.cap));
            if(d>0){
                e.cap-=d;
                G[e.to][e.rev].cap+=d;
                return d;
            }
        }
    }
    return 0;
}
int dinic(int s,int t){
    int flow=0;
    while(1){
        bfs(s);
        if(level[t]<0)return flow;
        memset(iter,0,sizeof(iter));
        int f;
        while((f=dfs(s,t,maxinf))>0)
            flow+=f;
    }
}
int main(){
    int t;cin>>t;
    while(t--){
        cin>>n>>m;
        for(int i=1;i<=m;i++){
            cin>>uu[i]>>vv[i]>>dd[i];
            mp[uu[i]].push_back(make(vv[i],dd[i]));
        }
        dijkstra(1,dis1);
//        for(int i=1;i<=n;i++){
//            cout<<dis1[i]<<endl;
//        }
        for(int i=1;i<=n;i++)
            mp[i].clear();
        for(int i=1;i<=m;i++)
            mp[vv[i]].push_back(make(uu[i],dd[i]));
        dijkstra(n,dis2);
        for(int i=1;i<=m;i++){
            if(dis1[uu[i]]+dd[i]+dis2[vv[i]]==dis1[n]){
//                cout<<uu[i]<<" "<<vv[i]<<endl;
                add(uu[i],vv[i],dd[i]);
            }

        }
        cout<<dinic(1,n)<<endl;
        for(int i=1;i<=n;i++){
            G[i].clear();
            mp[i].clear();
        }
    }
}