【NOI2015模拟1.10】B组黑白树

Description

给定一棵树,边的颜色为黑或白,初始时全部为白色。维护两个操作:
1. 查询 u 到根路径上的第一条黑色边的标号。
2. 将 u 到 v 路径上的所有边的颜色设为黑色。

Input

第一行两个数 n, m 分别表示点数和操作数。
接下来 n − 1 行,每行 2 个数 u, v 表示一条 u 到 v 的边。
接下来 m 行,每行为以下格式:
• 1 v 表示第一个操作
• 2 v u 表示第二种操作

Output

对于每个询问,输出相应答案。如果不存在,输出 0 。

Sample Input

5 4
1 2
1 3
2 4
2 5
1 2
2 2 3
1 3
1 4

Sample Output

0
2
1

Data Constraint

对于 30% 的数据:n, m ≤ 10 3
对于 100% 的数据:n, m ≤ 10 6

Hint

注意：根是1

思路

先将所有操作正着进行一遍，将所有的黑边相邻的点按照关系合并，就是一个集合中的代表元素一定是深度最小的点。
然后找出所有自始至终都是白色的边，以及每条边变黑的时间。将白边用并查集合并
倒着做所有的操作，对于染黑操作如果我们撤销相当于染白，将是所有在当前操作中变黑的边的两端用并查集合并，可以直接遍历路径，做法与合并黑边时类似。（也可以按照每条边变黑的时间排序，然后直接合并每一条边，这样就不用遍历路径了）
对于每次的查询操作直接找集合的代表元素，代表元素与其父节点之间的边就是答案。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define N 1000333
using namespace std;
long long tot,p[N],a[N*2],v[N*2],c[N*2],fa[N],son[N],n,m,ma[N],sz,d[N],f[N],pd[N],pd1[N],ass[N];
struct A
{
    long long opt,x,y;
}e[N],t[N];
int read()  
 {  
    char ch = getchar();  
    for ( ; ch > '9' || ch < '0'; ch = getchar());  
    int tmp = 0;  
    for ( ; '0' <= ch && ch <= '9'; ch = getchar())  
      tmp = tmp * 10 + int(ch) - 48;  
    return tmp;  
 }   
void add(int x,int y,int num)
{
    tot++; a[tot]=p[x]; p[x]=tot; v[tot]=y; c[tot]=num;
    tot++; a[tot]=p[y]; p[y]=tot; v[tot]=x; c[tot]=num;
}
void dfs(int x,int f)
{
    d[x]=d[f]+1;
    for (int i=p[x];i;i=a[i]) 
    {
        if (v[i]==f) continue;
        int t=c[i];
        e[t].x=x; e[t].y=v[i];
        fa[v[i]]=x;
        dfs(v[i],x);
        ma[v[i]]=c[i];
    }
}
int find(int x)
{
    if (f[x]==x) return x;
    f[x]=find(f[x]);
    return f[x];
}
void ch(int x,int y,int opt)
{
    x=find(x); y=find(y);
    while (x!=y) 
    {
        if (d[x]<d[y]) swap(x,y);
        if (!pd[x]) f[x]=f[fa[x]],pd[x]=opt;
        x=f[x];
    }
}
void settle(int x,int y,int opt)
{
    x=find(x); y=find(y);
    while (x!=y) 
    {
        if (d[x]<d[y]) swap(x,y);
        if (pd[x]==opt) f[x]=f[fa[x]];
        x=fa[x];
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<n;i++) 
    {
        int x,y;
        //scanf("%d%d",&x,&y);
        x=read(); y=read();
        add(x,y,i);
    }
    dfs(1,0);
    for (int i=1;i<=n;i++) f[i]=i;
    for (int i=1;i<=m;i++) 
    {
        scanf("%d%d",&t[i].opt,&t[i].x);
        if (t[i].opt==2) scanf("%d",&t[i].y),ch(t[i].x,t[i].y,i);
    }
    for (int i=1;i<=n;i++) f[i]=i;
    for (int i=1;i<=n;i++) pd1[i]=pd[i];
    for (int i=2;i<=n;i++)
     if (!pd1[i]) 
     {
        int t=ma[i];
        int r1=find(e[t].x); int r2=find(e[t].y);
        f[r2]=r1;
     } 
    int cnt=0;
    for (int i=m;i>=1;i--) 
    {
        if (t[i].opt==1) 
        {
            int r1=find(t[i].x);
            ass[++cnt]=ma[r1];
        }
        else settle(t[i].x,t[i].y,i);
    }
    for (int i=cnt;i>=1;i--) printf("%d\n",ass[i]);
}