MnZn 求助 TLE on #9
查看原帖
MnZn 求助 TLE on #9
614725
masonpop楼主2023/5/23 17:09

感觉很离谱,自己照着网上一篇博客学的 dinic,感觉就算最坏 O(n2m)O(n^2m) 也远远不到 TLE 的地步。自测的 #9 跑了 8s 后跑出了正确答案,所以应该不是死循环的问题。求大佬解答,是常数问题还是哪里写假了。

#include <bits/stdc++.h>
using namespace std;
#define Mem(a) memset((a),0,sizeof(a))
#define int long long
const int inf=1e15;
const int maxn=4e4+10;
int n,m,s,t,head[maxn],to[maxn],nxt[maxn],ver[maxn],tot;
int dep[maxn];
//flow:到当前点1能增广的最大流量
//dep:深度,用于分层 
int ans;
queue<int> q;
inline void add(int x,int y,int z)
{
	to[++tot]=y;
	nxt[tot]=head[x];
	head[x]=tot;
	ver[tot]=z;
}
inline int rev(int x)//反边
{
	if(x&1)return x+1;
	return x-1;	
} 
int flag;
inline void bfs()
{
	memset(dep,0,sizeof(dep));
	while(!q.empty())q.pop();//清空
	q.push(s);dep[s]=1;
	while(!q.empty())
	{
		int x=q.front();q.pop();
		for(int i=head[x];i;i=nxt[i])
		{
			int y=to[i];
			if(dep[y] || !ver[i])continue;//访问过或者枯竭了
			dep[y]=dep[x]+1; 
			q.push(y); 
		}
	}
}
inline int dfs(int x,int flow)
{
	if(x==t)
	{
		ans+=flow;
		flag=1;
		return flow;
	}
	for(int i=head[x];i;i=nxt[i])
	{
		int y=to[i];
		if(dep[y]!=dep[x]+1 || !ver[i])continue;//层数不对或枯竭了
		int delt=dfs(y,min(flow,ver[i]));
		if(flag)
		{
			ver[i]-=delt;
			ver[rev(i)]+=delt;
			return delt;
		}
	}
	return 0;
}
inline void dinic()
{
	while(1)
	{
		bfs();
		if(!dep[t])return;//到不了
		while(1)
		{
			flag=0;
			dfs(s,inf);
			if(!flag)break; 
		}
	}
}
signed main()
{
	scanf("%lld%lld%lld%lld",&n,&m,&s,&t);
	for(int i=1;i<=m;i++)
	{
		int x,y,z;
		scanf("%lld%lld%lld",&x,&y,&z);
		add(x,y,z);add(y,x,0);
	}
	dinic();
	printf("%lld\n",ans);
	return 0;
}
2023/5/23 17:09
加载中...