[agc14e]Blue and Red Tree

前言

这题主要是要想到倒过来，是一个很唯一的过程。
我大概想到了，但我比题解蠢。
题解的不需要区分红蓝让我觉得很机智。

题意

有一颗全是蓝边的树，你对其执行n-1次操作。
每次操作选择一条全是蓝边的路径，将其中的一条蓝边断开，假设选择的路径是j->k，断的边是x-y，且断开后j和x联通，那么k和y联通。
然后你需要选择j->x上的一个节点，以及k->y上的一个节点，在这两个节点间连红边。
现在问你能否存在方案变着指定的全是红边的树。

题解

考虑倒过来加边。
这个过程是很唯一的。
首先一开始有很多单点联通块。
每次如果两个联通块之间存在蓝边与红边相连，合并两个联通块。
最后合并成一个就行了。
好实现的方法是，这个红和蓝我们可以认为没有区别。
于是可以用一个queue存需要合并的联通块对（即之间存在两条边），用map来存两个联通块间的边数，用set存与某个联通块存在边的联通块，用并查集来维护连通性，合并两个联通块只需要启发式合并即可。
这样真好写！

#include<cstdio>#include<algorithm>#include<set>#include<map>#include<queue>#define fo(i,a,b) for(i=a;i<=b;i++)using namespace std;const int maxn=100000+10;typedef pair<int,int> pi;pi zlt;map<pi,int> cnt,bz;set<int> s[maxn];set<int>::iterator it;queue<pi> dl;int fa[maxn];int i,j,k,l,t,n,m,x,y,z,ca;bool czy;int getfa(int x){    return fa[x]?fa[x]=getfa(fa[x]):x;}int main(){    scanf("%d",&n);    fo(i,1,2*(n-1)){        scanf("%d%d",&j,&k);        if (j>k) swap(j,k);        zlt=make_pair(j,k);        cnt[zlt]++;        if (cnt[zlt]==1){            s[j].insert(k);            s[k].insert(j);        }        if (cnt[zlt]>1&&!bz[zlt]){            dl.push(zlt);            bz[zlt]=1;        }    }    czy=1;    ca=n-1;    while (ca--){        x=y=-1;        while (!dl.empty()){            j=getfa(dl.front().first);            k=getfa(dl.front().second);            dl.pop();            if (j!=k){                x=j;y=k;                break;            }        }        if (x==y&&x==-1){            czy=0;            break;        }        if (s[x].size()<s[y].size()) swap(x,y);        fa[y]=x;        it=s[y].begin();        while (it!=s[y].end()){            z=*it;            it++;            s[z].erase(s[z].find(y));            if (z==x) continue;            //if (s[x].find(z)==s[x].end()) s[x].insert(z),s[z].insert(x);            if (x<z) zlt=make_pair(x,z);else zlt=make_pair(z,x);            cnt[zlt]++;            if (cnt[zlt]==1){                s[x].insert(z);                s[z].insert(x);            }            if (cnt[zlt]>1&&!bz[zlt]){                dl.push(zlt);                bz[zlt]=1;            }        }        s[y].clear();    }    if (czy) printf("YES\n");else printf("NO\n");}