感觉很离谱,自己照着网上一篇博客学的 dinic,感觉就算最坏 O(n2m) 也远远不到 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;
}