有无好心人帮忙调dinic,全部TLE
查看原帖
有无好心人帮忙调dinic,全部TLE
574916
SunsetLake楼主2023/12/20 09:30

rt,dinic应该没写错,粘了板子后还是TLE。

#include<bits/stdc++.h>
#define ll long long
#define pb push_back
using namespace std;
const int N=306,inf=1e9;
int T,n,m,k,cnt;
int head[N],dis[N],cur[N];
int a[N],b[N];
int s,t;
struct edge{
	int u,v,nxt;
	int w;
}e[N*100];
void init(){
	s=0;t=2*n+1;
	k=1;
	for(int i=1;i<=t;++i)head[i]=dis[i]=cur[i]=0;
}
void add(int u,int v,int w){
	e[++k].v=v;
	e[k].nxt=head[u];
	e[k].w=w;
	head[u]=k;
} 
queue<int>q;
bool bfs(){
	while(!q.empty())q.pop();
	for(int i=1;i<=t;++i)dis[i]=inf;
	q.push(s);cur[s]=head[s];
	while(!q.empty()){
		int u=q.front();
		q.pop();
		for(int i=head[u];i;i=e[i].nxt){
			int v=e[i].v;
			if(dis[v]==inf&&e[i].w){
				dis[v]=dis[u]+1;
				cur[v]=head[v];
				if(v==t)return 1;
				q.push(v);
			}
		}
	}
	return 0;
}
int dfs(int u,int sum){
	if(u==t)return sum;
	int res=0;
	for(int i=cur[u];i&&sum;i=e[i].nxt){
		int v=e[i].v;
		cur[u]=i;
		if(dis[v]==dis[u]+1&&e[i].w){
			int tmp=dfs(v,min(e[i].w,sum));
			if(!tmp)dis[v]=inf;
			e[i].w-=tmp;
			e[i^1].w+=tmp;
			res+=tmp;
			sum-=tmp;
		}
	}
	return res;
}
int dinic(){
	int res=0;
	while(bfs())res+=dfs(s,inf);
	return res;
}
void solve(){
	cin>>n;
	init();
	int cnt=0;
	for(int i=1;i<=n;++i)cin>>a[i];
	for(int i=1;i<=n;++i)cin>>b[i];
	for(int i=1;i<=n;++i){
		if((!a[i])||(!b[i]))add(s,i,1),add(i,s,0),cnt++;
		if(a[i])add(i+n,t,1),add(t,i+n,0);
	}
	for(int i=1;i<=n;++i){
		int x;
		for(int j=1;j<=n;++j){
			cin>>x;
			if(x||(i==j))add(i,j+n,1),add(j+n,i,0);
		}
	}
	int ol=dinic();
	cout<<(ol==cnt?"^_^":"T_T")<<endl;
}
int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);cout.tie(0);
	cin>>T;
	while(T--)solve();
	return 0;
}

2023/12/20 09:30
加载中...