求助:图一样,网络流板子一样,跑网络流的时间不一样
查看原帖
求助:图一样,网络流板子一样,跑网络流的时间不一样
623636
Thomas0702楼主2025/7/1 22:19

按照第一篇题解的思路,敲出了如下代码。

#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define lll __int128
#define db double
#define ld long double
#define eps 1e-8
#define fir first
#define sec second
#define pb push_back
#define ppc __builtin_popcount
using namespace std;

const int maxn=16400,inf=1e9;

int N,a[130][130],c[130][130];

inline int g(int i,int j){return (i-1)*N+j;}

namespace FLOW{
struct edge{int v,w,prev;}E[maxn*10];
int S,T,head[maxn<<1],cnt;
inline void add(int u,int v,int w){E[++cnt]={v,w,head[u]};head[u]=cnt;}
inline void add_edge(int u,int v,int w){add(u,v,w),add(v,u,0);}
int now[maxn<<1],d[maxn<<1];
bool bfs(){
	for(int i=0;i<=T;i++) d[i]=0,now[i]=head[i];
	queue<int> q;
	q.push(S);
	d[S]=1;
	while(!q.empty()){
		int u=q.front();q.pop();
		for(int i=head[u],v=E[i].v;i;v=E[i=E[i].prev].v)
			if(E[i].w&&!d[v])
				d[v]=d[u]+1,q.push(v);
	}
	return d[T];
}
inline int dfs(int u,int flow){
	if(u==T||!flow) return flow;
	int val=0;
	for(int i=now[u],v=E[i].v;i;v=E[i=E[i].prev].v){
		now[u]=i;
		if(!E[i].w||d[v]!=d[u]+1) continue;
		int t=dfs(v,min(flow,E[i].w));
		val+=t,flow-=t;
		E[i&1?i+1:i-1].w+=t,E[i].w-=t;
		if(!flow) break;
	}
	if(!val) d[u]=0;
	return val;
}
int dinic(){
	int ans=0,t;
	while(bfs())
		while((t=dfs(S,inf))>0)
			ans+=t;
	return ans;
}
}//namespace FLOW
using FLOW::add_edge;
using FLOW::S;
using FLOW::T;

mt19937 RND(time(0));
inline int rnd(int l,int r){return RND()%(r-l+1)+l;}
void gen(){
	N=128;
	for(int i=1;i<=N;i++)
		for(int j=1;j<=N;j++)
			a[i][j]=(i+j-2)%N+1;
	for(int i=1;i<=N;i++){
		int x=rnd(1,N),y=rnd(1,N);
		for(int j=1;j<=N;j++) swap(a[x][j],a[y][j]);
		x=rnd(1,N),y=rnd(1,N);
		for(int j=1;j<=N;j++) swap(a[j][x],a[j][y]);
	}
	cerr<<N<<"\n";
	for(int i=1;i<=N;i++)
		for(int j=1;j<=N;j++)
			c[i][j]=rnd(0,1);
	for(int i=1;i<=N;i++,cerr<<"\n")
		for(int j=1;j<=N;j++)
			cerr<<a[i][j]<<" ";
	for(int i=1;i<=N;i++,cerr<<"\n")
		for(int j=1;j<=N;j++)
			cerr<<c[i][j]<<" ";
}

int main(){
	freopen("log.txt","w",stderr);
//	gen();
	cin>>N;
	for(int i=1;i<=N;i++)
		for(int j=1;j<=N;j++)
			cin>>a[i][j];
	for(int i=1;i<=N;i++)
		for(int j=1;j<=N;j++)
			cin>>c[i][j];

	S=0,T=2*N*N+1;
	for(int i=1;i<=N;i++){
		for(int j=1;j<=N;j++){
			int u=g(i,j);
			add_edge(u,u+N*N,1-c[i][j]+2*N);
			if(a[i][j]==1) add_edge(S,u,inf);
			if(a[i][j]==N) add_edge(u+N*N,T,inf);
			for(int k=1;k<=N;k++){
				if(a[i][k]==a[i][j]+1) add_edge(u+N*N,g(i,k),inf);
				if(a[k][j]==a[i][j]+1) add_edge(u+N*N,g(k,j),inf);
			}
		}
	}

	int flow=FLOW::dinic();
//	cerr<<"flow = "<<flow<<"\n";
	printf("%d\n",N-flow+2*N*N);
	return 0;
}

会 TLE on 12。

但是如果将 108 至 111 行换成

			for(int k=1;k<=N;k++)
				if(a[i][k]==a[i][j]+1)
					add_edge(u+N*N,g(i,k),inf);
			for(int k=1;k<=N;k++)
				if(a[k][j]==a[i][j]+1)
					add_edge(u+N*N,g(k,j),inf);

就过了,而且跑的飞快。

这是为什么呢?

2025/7/1 22:19
加载中...