AGC 041E - Balancing Network

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

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

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

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

代码

#include
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];
    queueQ;
    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 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();
    }
}

你可能感兴趣的:(AGC 041E - Balancing Network)