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();
}
}