Description
给定两棵树,第一颗树边都为蓝色,在这棵树你可以进行若干次以下操作:选择一条只包含蓝色边的路径,删除路径上一条边,然后路径的两个端点连一条红色的边。问是否能使得第一棵树与第二棵树完全相同(形态相同且点的编号互相对应)。
Solution
考虑最后删除的一定是两棵树同时出现的边,于是我们把这些边的两个端点缩成一个点,表示这些边不能先删除。于是我们发现新树变成了一个同样的问题,同样把最后删除的边的两个端点缩起来。最终,如果我们发现最后变成一个点,则说明可行,否则就不可行。
具体的话可以把两棵树的边都加入,每次选择有两条边相连的两个点,启发式合并一下。
Code
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#define fo(i,j,k) for(int i=j;i<=k;++i)
#define fd(i,j,k) for(int i=j;i>=k;--i)
using namespace std;
const int N=1e5+10;
typedef pair<int,int> pr;
map<pr,int> mp;
multiset<int> b[N];
queue<pr> q;
typedef multiset<int>::iterator it;
int f[N];
int find(int x){return !f[x]?x:f[x]=find(f[x]);
}
void link(int x,int y){b[x].insert(y),b[y].insert(x);
}
pr mkpr(int x,int y){if(x>y) swap(x,y);return make_pair(x,y);
}
int main()
{int n;scanf("%d",&n);fo(i,1,n*2-2){int x,y;scanf("%d %d",&x,&y);link(x,y);pr t=mkpr(x,y);++mp[t];if(mp[t]==2) q.push(t);if(mp[t]>2) return printf("NO"),0;}fo(ch,2,n){if(q.empty()) return printf("NO"),0;int x=0,y=0;for(;x==y;x=find(q.front().first),y=find(q.front().second),q.pop());if(b[x].size()>b[y].size()) swap(x,y);f[x]=y;for(it i=b[x].begin();i!=b[x].end();i++){int v=find(*i);b[v].erase(b[v].find(x));if(v==y) continue;link(v,y);pr t=mkpr(v,y);++mp[t];if(mp[t]==2) q.push(t);}b[x].clear();}printf("YES");
}