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

又是一道花了一天才过的题
就感觉网络流的题都很神奇 莫名其妙的就变成网络流了
说到底还是我太菜了
一开始建了双向图跑最短路 一直wa
找了好久才过 头疼得很

题目思路

题目要求要最快到达并且每个城市都只能跑一次 问男孩能到女孩那几次
做法就是先建图跑最短路 找到符合最短路的边
再把这些边重新建一个图 因为这些边都在最短路上
所以拿这些边跑最大流 一定符合题目条件
可以只跑一次最短路 也可以正反建两次图跑最短路
区别就是判定边的时候
第一个是满足 dis[v] == dis[u]+e[i].v
第二个要满足dis1[u]+dis2[v]+e[i].v == dis1[t]
两种都行 主要第一个只用建一次图 所以我用的第一种

ac代码

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <vector>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <utility>
#define pi 3.1415926535898
#define ll long long
#define lson rt<<1
#define rson rt<<1|1
#define eps 1e-6
#define ms(a,b) memset(a,b,sizeof(a))
#define legal(a,b) a&b
#define print1 printf("111\n")
using namespace std;
const int maxn = 2e6+10;
const int inf = 0x3f3f3f3f;
const ll llinf =0x3f3f3f3f3f3f3f3f;
const int mod = 1e9+7;

struct node
{
    int to,next,v;
}e[maxn],edge[maxn];
int n,m,first[maxn],ffirst[maxn],dis[maxn];
int vis[maxn],len1,len2,d[maxn];

void add(int u,int v,int w)
{
    e[len1].to=v;
    e[len1].next=first[u];
    e[len1].v=w;
    first[u]=len1++;


}

void Add(int u,int v,int w)
{
    edge[len2].to=v;
    edge[len2].v=w;
    edge[len2].next=ffirst[u];
    ffirst[u]=len2++;
    edge[len2].to=u;
    edge[len2].v=0;
    edge[len2].next=ffirst[v];
    ffirst[v]=len2++;
}

struct point
{
    int id,val;
    point(int id,int val)
    {
        this->id=id;
        this->val=val;
    }
    bool operator<(const point&x)const
    {
        return val>x.val;
    }
};

void dij(int s)
{
    //ms(vis,0);
    for(int i=0;i<=n+1;i++)
        dis[i]=inf,vis[i]=0;
    priority_queue<point>q;
    q.push(point(s,0));
    dis[s]=0;
    while(!q.empty())
    {
        int rt=q.top().id;
        q.pop();
        if(vis[rt])continue;
        vis[rt]=1;
        for(int i=first[rt];i!=-1;i=e[i].next)
        {
            int id=e[i].to;
            if(!vis[id]&&dis[rt]+e[i].v<dis[id])
            {
                dis[id]=dis[rt]+e[i].v;
                q.push(point(id,dis[id]));
            }
        }
    }
}

void init()
{
    len1=len2=0;
    ms(first,-1);
    ms(ffirst,-1);
}

bool makelevel(int s,int t)
{
    ms(d,0);
    queue<int>q;
    d[s]=1;
    q.push(s);
    while(!q.empty())
    {
        int x=q.front();
        q.pop();
        if(x==t)return true;
        for(int i=ffirst[x];i!=-1;i=edge[i].next)
        {
            if(d[edge[i].to]==0&&edge[i].v!=0)
            {
                q.push(edge[i].to);
                d[edge[i].to]=d[x]+1;
            }
        }
    }
    return false;
}

int dfs(int x,int flow,int t)
{
    if(x==t)return flow;
    int sum=0;
    for(int i=ffirst[x];i!=-1;i=edge[i].next)
    {
        if(edge[i].v!=0&&d[edge[i].to]==d[x]+1)
        {
            int tem=dfs(edge[i].to,min(flow-sum,edge[i].v),t);
            edge[i].v-=tem;
            edge[i^1].v+=tem;
            sum+=tem;
            if(sum==flow)return sum;
        }
    }
    return sum;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        init();
        scanf("%d%d",&n,&m);
        for(int i=1;i<=m;i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
        }
        int s,t;
        scanf("%d%d",&s,&t);
        dij(s);
        for(int u=1;u<=n;u++)
        {
            for(int i=first[u];i!=-1;i=e[i].next)
            {
                int id=e[i].to;
                if(dis[id]==dis[u]+e[i].v)
                {
                    //print1;
                    Add(u,id,1);
                }
            }
        }
        int ans=0;
        while(makelevel(s,t))
        {
            //print1;
            ans+=dfs(s,inf,t);
        }
        printf("%d\n",ans);
    }
    return 0;
}
;