dinic 10pts 点1-10MLE求助
查看原帖
dinic 10pts 点1-10MLE求助
382274
暗影之梦楼主2022/3/18 18:32
#include<iostream>
#include<queue>
#include<cstring>
#define int long long
using namespace std;
inline int read()
{
	int x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')
        {
            f=-1;
        }
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
inline void write(int x)
{
    if(x<0)
    {
        putchar('-');
        x=-x;
    }
    if(x>9)
    {
        write(x/10);
        putchar(x%10+'0');
        return;
    }
    putchar(x+'0');
} 
int n,m,s,t,head[100001],cnt,dep[2001],nwb[2001];
struct edge1
{
	int to,nxt,w;
}e[100001];
void edge(int u,int v,int w)
{
	e[++cnt].to=v;
	e[cnt].w=w;
	e[cnt].nxt=head[u];
	head[u]=cnt;
}
int dfs(int nw,int w)
{
	if(nw==t||w==0) return w;
	for(int &i=nwb[nw];i;i=e[i].nxt)
	{
		if(dep[e[i].to]==dep[nw]+1&&e[i].w);
		{
			int wi=dfs(e[i].to,min(w,e[i].w));
			if(wi>0)
			{
				e[i].w-=wi;
				e[i^1].w+=wi;
				return wi;
			}
		}
	}
	return 0;
}
int bfs()
{
	queue<int> q;
	memset(dep,-1,sizeof(dep));
	dep[s]=1;
	q.push(s);
	while(!q.empty())
	{
		int nw=q.front();
		q.pop();
		for(int i=head[nw];i;i=e[i].nxt)
		{
			if(e[i].w&&dep[e[i].to]==-1)
			{
				dep[e[i].to]=dep[nw]+1;
				q.push(e[i].to);
				if(e[i].to==t) return 1;
			}
		}
	}
	return 0;
}
int dinic()
{
	int ans=0;
	while(bfs())
	{
//		cout<<114514<<endl;
		for(int i=1;i<=n;i++) nwb[i]=head[i];
		while(int ans1=dfs(s,1e16)) ans+=ans1;
	}
	return ans;
}
signed main()
{
	n=read(),m=read(),s=read(),t=read();
	for(int i=1;i<=m;i++) 
	{
		int u=read(),v=read(),w=read();
		edge(u,v,w);
		edge(v,u,0); 
	}
	write(dinic());
	return 0;
}

dfs和bfs出了问题但找不到,求大佬们帮忙看看

2022/3/18 18:32
加载中...