爆零求助
查看原帖
爆零求助
793625
Pink_Cut_Tree楼主2024/7/12 17:29
#include<bits/stdc++.h>
using namespace std;
const int N=5e5+5,inf=2e9;
int T;
int s,t=1002;
int n,a[N],b[N];
int head[N],cnt,tot;
int dep[N],cur[N];
struct Edge{
	int to,nxt,value;
}edges[N];
void add(int u,int v,int w){
	edges[++cnt]={v,head[u],w};
	head[u]=cnt;
}
bool bfs(){
	for(int i=1;i<=t;i++){
		dep[i]=inf;
	}
	memcpy(cur,head,sizeof cur);
	queue<int>Q;
	Q.push(s);
	while(!Q.empty()){
		int u=Q.front(); Q.pop();
		for(int i=head[u];i;i=edges[i].nxt){
			int v=edges[i].to;
			if(dep[v]>dep[u]+1&&edges[i].value){
				dep[v]=dep[u]+1;
				Q.push(v);
			}
		}
	}
return dep[t]!=inf;
}
int max_flow;
int dfs(int u,int flow){
	if(u==t){
		max_flow+=flow;
		return flow;
	}
	int used=0;
	for(int i=cur[u];i;i=edges[i].nxt){
		cur[u]=i;
		int v=edges[i].to;
		if(dep[v]==dep[u]+1&&edges[i].value){
			int rflow=dfs(v,min(flow-used,edges[i].value));
			if(rflow){
				used+=rflow;
				edges[i].value-=rflow;
				edges[i^1].value+=rflow;
				if(flow==used){
					return used;
				}
			}
		}
	}
return used;
}
int dinic(){
	while(bfs()){
		dfs(s,inf);
	}
return max_flow;
}
bool flag;
int main(){
	cin.tie(0)->sync_with_stdio(0);
	cin>>T;
	while(T--){
		cin>>n;
		memset(cur,0,sizeof cur);
		memset(head,0,sizeof head);
		memset(dep,0,sizeof dep);
		cnt=1,tot=0;
		for(int i=1;i<=n;i++){
			cin>>a[i];
			if(a[i]){
				add(i+n,t,1);
				add(t,i+n,0);
			}
		} 
		for(int i=1;i<=n;i++){
            cin>>b[i];
            if((a[i]&&!b[i])||(a[i]==0)){
                add(s,i,1);
                add(i,s,0);
                tot++;
            }
        }
        for(int i=1;i<=n;i++){
        	for(int j=1;j<=n;j++){
        		cin>>flag;
        		if(flag||i==j){
        			add(i,j+n,1);
        			add(j+n,i,0);
				}
			}
		}
		if(dinic()>=tot){
			cout<<"^_^\n";
		}
		else{
			cout<<"T_T\n";
		}
	}
return 0;
}
2024/7/12 17:29
加载中...