HDU-3416 Marriage Match IV(最短路+最大流)

HDU-3416

n个点m条带权有向边，求起点到终点的最短路径的条数

先dij求出起点单源最短路，网络流中根据距离关系确定这条边是否可以通过流 即$dist[j]==dist[i]+wt[i,j]$,没选相当于删边。
当然也可以求两次单源最短路，给起点终点各求一次，根据$dist1[i]+dist2[j]+wt[i,j]==dist1[t]$，$dist1,dist2$分别为起点终点单源最短路,这个好像快一点。

/*
    开始是通过Dij删边， tle 3
    提示了最大流后才想到
    先dij求出起点单源最短路，网络流中根据距离关系确定这条边是否可以通过流 即dist[j]==dist[i]+wt[i,j],没选相当于删边
    当然也可以求两次单源最短路，给起点终点各求一次，根据dist1[i]+dist2[j]+wt[i,j]==dist1[t] ，dist1,dist2分别为起点终点单源最短路,这个好像快一点
    一开始把增广路的边权设为INF,没有考虑增广路的返流 wa3
*/
#include <queue>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;
typedef pair<int,int> PII;
const int MAXN=1e3+7;
const int MAXM=1e5+7;
const int INF=1e9+7;
int tot,head[MAXN],pre[MAXN],mic[MAXN],di[MAXN],n,m,iter[MAXN],num[MAXN];
int dist[MAXN];
struct edge
{
    int ne,from,to,cap,wt;
}e[MAXM<<1];
void init()
{
    tot=0;
    memset(head,-1,sizeof(head));
}
void addedge(int a,int b,int c)
{
    e[tot]=(edge){head[a],a,b,1,c};
    head[a]=tot++;
    e[tot]=(edge){head[b],b,a,0,-c};
    head[b]=tot++;
}
int vis[MAXN];
priority_queue<PII> pq;
int dijkstra(int s,int t)//s为源点，t为终点，路径权值非负
{
    for(int i=1;i<=n;i++)
        dist[i]=INF,vis[i]=0;
    dist[s]=0;
    pq.push(PII(0,s));
    while(!pq.empty())
    {
        int milb=pq.top().second;
        pq.pop();
        if(vis[milb])
            continue;
        vis[milb]=1;
        for(int j=head[milb];~j;j=e[j].ne)
        {
            int v=e[j].to;
            if(e[j].cap&&!vis[v]&&dist[v]>dist[milb]+e[j].wt)
            {
                dist[v]=dist[milb]+e[j].wt;
                pq.push(PII(-dist[v],v));
            }
        }
    }
    return dist[t];
}
int isap(int s,int t,int totn)//点标号范围为1～totn 源点s 汇点t
{
    queue<int> q;
    int p,adv,res=0;
    memset(di,-1,sizeof(di));
    di[t]=0;
    q.push(t);
    while(!q.empty())
    {
        p=q.front();
        q.pop();
        for(int i=head[p];~i;i=e[i].ne)
            if(e[i^1].cap>0&&!~di[e[i].to])
            {
                di[e[i].to]=di[p]+1;
                q.push(e[i].to);
            }
    }
    for(int i=1;i<=totn;i++)
    {
        iter[i]=head[i];
        if(~di[i])
            num[di[i]]++;
    }
    int x=s;
    mic[s]=INF;
    pre[s]=-1;
    while(di[s]<totn)
    {
        if(x==t)
        {
            int cur=pre[t],ff=mic[t];
            while(~cur)
            {
                e[cur].cap-=ff;
                e[cur^1].cap+=ff;
                cur=pre[e[cur].from];
            }
            res+=ff;
            x=s;
        }
        adv=0;
        for(int& i=iter[x];~i;i=e[i].ne)
            if(dist[e[i].to]==dist[x]+e[i].wt&&di[x]==di[e[i].to]+1&&e[i].cap>0)
            {
                int v=e[i].to;
                adv=1;
                pre[v]=i;
                mic[v]=min(mic[x],e[i].cap);
                x=v;
                break;
            }
        if(!adv)
        {
            int m=totn-1;
            for(int i=head[x];~i;i=e[i].ne)
                if(dist[e[i].to]==dist[x]+e[i].wt&&e[i].cap>0)
                    m=min(m,di[e[i].to]);
            if(--num[di[x]]==0)//gap优化
                break;
            num[di[x]=m+1]++;
            iter[x]=head[x];
            if(x!=s)
                x=e[pre[x]].from;
        }
    }
    return res;
}
int main()
{
    int ta,tb,tc,T;
    scanf("%d",&T);
    for(int kase=1;kase<=T;kase++)
    {
        init();
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&ta,&tb,&tc);
            addedge(ta,tb,tc);
        }
        scanf("%d%d",&ta,&tb);
        int ans=0;
        if(dijkstra(ta,tb)==INF)
            printf("0\n");
        else
            printf("%d\n",isap(ta,tb,n));
    }
    return 0;
}