当前位置: 代码迷 >> 综合 >> HDU 3416 Marriage Match IV(最短路dij + 最大流dinic)
  详细解决方案

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

热度:36   发布时间:2024-02-01 04:21:06.0

又是一道花了一天才过的题
就感觉网络流的题都很神奇 莫名其妙的就变成网络流了
说到底还是我太菜了
一开始建了双向图跑最短路 一直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;
}
;
  相关解决方案