AGC 041E - Balancing Network

可以对应到一个 \(\text{DAG}\) 上:

  • 对于一条原来的操作,分别向晚于当前操作的、含有两端的最早的两个操作连边
  • 对于每个初始点,向含有这个位置的最早操作连边。
  • 对于每个点(操作),选择一条边作为出边。

如此,我们只需要找到一种方式所有初始点在同一颗树里,或不在同一树里。

第一问可以暴力bitset解决,而第二问可以发现相当于每个点经过两次,这等价于一个网络流,而我们只需要增广两次。

代码

#include<bits/stdc++.h>
using namespace std;

const int N = 5e4+5, M = 1e6+5;
int n,m;

int hed[M],cnt=1,to[M],nxt[M],cap[M],flow[M];
void adde(int u,int v,int _cap,int tpe=0){
	++cnt;to[cnt]=v,cap[cnt]=_cap,nxt[cnt]=hed[u],flow[cnt]=0;hed[u]=cnt;
	if(!tpe){++cnt;to[cnt]=u,cap[cnt]=0,nxt[cnt]=hed[v],flow[cnt]=0;hed[v]=cnt;}
}
namespace graph{
	int S,T;
	int d[M];bool vis[M];
	queue<int>Q;
	bool bfs(){
		memset(vis,0,sizeof(vis));memset(d,0,sizeof(d));
		Q.push(S);vis[S]=1;
		while(!Q.empty()){
			int u=Q.front();Q.pop();
			for(int i=hed[u];i;i=nxt[i]){
				int v=to[i];
				if(!vis[v]&&cap[i]>flow[i]){
					vis[v]=1,d[v]=d[u]+1;
					Q.push(v);
				}
			}
		}
		return vis[T];
	}
	int dfs(int u,int F){
		if(u==T||!F)return F;
		int flownow=0;
		for(int i=hed[u];i;i=nxt[i]){
			int v=to[i];if(d[v]!=d[u]+1)continue;
			int f=dfs(v,min(F,cap[i]-flow[i]));
			flownow+=f,F-=f,flow[i]+=f,flow[i^1]-=f;
			if(!F)break;
		}
		if(!flownow)d[u]=0;
		return flownow;
	}
	const int inf=0x3f3f3f3f;
	int max_flow(){
		int fw=0;
		while(bfs()){
			fw+=dfs(S,inf);
			if(fw>1)break;
		}
		return fw;
	}
}


int nd[M],col[M];

namespace Subtask_1{
	bitset<N> last[N];
	int u[M],v[M];
	bool vis[M];
	inline void mdfs(int x){
		for(int i=hed[x];i;i=nxt[i]){
			int vv=to[i];
			if(!vis[vv]){
				vis[vv]=1;
				mdfs(vv);
				if(v[vv]==u[x]||v[vv]==v[x])col[vv]=1;
			}
		}
	}
	inline void solve(){
		int flg=0;
		for(int i=1;i<=n;i++)last[i].set(i);
		for(int i=1;i<=m;i++){
			scanf("%d%d",&u[i],&v[i]);
			adde(i,nd[u[i]],0,1),adde(i,nd[v[i]],0,1);
			last[u[i]] = last[v[i]] = last[u[i]]|last[v[i]];
			nd[u[i]]=nd[v[i]]=i;
			if(last[u[i]].count()==(unsigned)n){
				mdfs(i);
				flg=1;
				goto END;
			}
		}
		END:;
		if(!flg)puts("-1");
		else{
			for(int i=1;i<=m;i++){
				printf("%c",col[i]?'v':'^');
			}puts("");
		}
		return;
	}
}

int ed[M];
int E[M][2];
namespace Subtask_2{
	int u[M],d[M];
	inline void solve(){
		graph::S=m*2+1,graph::T=m*2+2;
		for(int i=1;i<=m;i++){
			scanf("%d%d",&u[i],&d[i]);
		}
		for(int i=m;i;i--){
			adde(i,m+i,1);
			if(!nd[u[i]])adde(m+i,graph::T,1),ed[u[i]]=i;
			else adde(m+i,nd[u[i]],1);
			if(!nd[d[i]])adde(m+i,graph::T,1),ed[d[i]]=i;
			else adde(m+i,nd[d[i]],1);
			E[i][0]=nd[u[i]],E[i][1]=nd[d[i]];
			nd[u[i]]=nd[d[i]]=i;
		}
		for(int i=1;i<=n;i++){
			if(!nd[i])adde(graph::S,graph::T,1);
			else adde(graph::S,nd[i],1);
		}
		int ret=graph::max_flow();
		if(ret<=1){puts("-1");return ;}
		for(int i=m+1;i<=m*2;i++){
			for(int e=hed[i];e;e=nxt[e])if(cap[e]>0&&flow[e]>0){
				int v=to[e];
				if(v==m*2+2){
					if(ed[u[i-m]]==i-m)col[i-m]=0;
					else col[i-m]=1;
					continue;
				}else if(E[i-m][1]==v)col[i-m]=1;
			}
		}
		for(int i=1;i<=m;i++){
			printf("%c",col[i]?'v':'^');
		}puts("");
	}
}

int T;
int main()
{
	cin >> n >> m >> T;
	if(T==1){
		Subtask_1::solve();
	}else{
		Subtask_2::solve();
	}
}

posted @ 2020-01-10 14:07  jerome_wei  阅读(445)  评论(0编辑  收藏  举报