[并查集][bfs]JZOJ P3973 黑白数

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

题解

• 首先，我们先将边打进队列里
• 那么，就可以一波bfs，做并查集，求出深度，位于哪条边
• 然后，就比较暴力了
• 读入，如果为"1"，那就一直往上跑，找到第一条出现的黑边
• 如果为"2",现将路径一直往上条，沿路都为黑边

代码

#include<cstdio>
#include<iostream>
using namespace std;
int n,m,x,y,z,tot,head,tail,ans,w[2000001],state[2000001],visit[2000001],deep[2000001],k[2000001],next[2000001],v[2000001],to[2000001],last[2000001],fa[2000001];
void insert(int x,int y,int z)
{
    next[++tot]=y;
    v[tot]=z;
    to[tot]=last[x];
    last[x]=tot;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n-1;i++)
    {
        scanf("%d%d",&x,&y);
        insert(x,y,i); insert(y,x,i);
    }
    head=0; tail=1; state[1]=1; visit[1]=1; deep[1]=1;
    while (head<tail)
    {
        head++; x=state[head];
        for (int i=last[x];i;i=to[i])
        {
            y=next[i];
            if (visit[y]==0)
            {
                fa[y]=x;
                deep[y]=deep[x]+1;
                k[y]=v[i];
                visit[y]=1;
                tail++;
                state[tail]=y;
            }
        }
    }
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        if (x==2)
        {
            scanf("%d",&z);
            if (deep[y]<deep[z]) swap(z,y);
            while (deep[y]>deep[z]) w[y]=1,y=fa[y];
            while (y!=z)
            {
                w[y]=w[z]=1;
                y=fa[y];
                z=fa[z];
            }
        }
        else 
        {
            ans=0;
            while (y!=0)
            {
                if (w[y]==1) { ans=k[y]; break; }
                y=fa[y];
            }
            printf("%d\n",ans);
        }
    }
    return 0;
}