CodeForces 466E. Information Graph (并查集+倍增)

最新推荐文章于 2022-07-24 12:03:36 发布

原创最新推荐文章于 2022-07-24 12:03:36 发布 · 429 阅读

0 ·

CC 4.0 BY-SA版权

并查集同时被 2 个专栏收录

10 篇文章

订阅专栏

倍增

4 篇文章

订阅专栏

本文解析了CodeForces上的一道题目，介绍了如何使用并查集和倍增算法处理员工间的上下级关系及文件签署查询。通过路径压缩优化查询效率，实现在线变离线的算法策略。

题目： http://codeforces.com/problemset/problem/466/E

题意：
三种操作：
1.y变为x的boss；
2.x签署一份文件并逐级往上传并签署；
3.查询第x个人是否签署第i份文件。

分析：
在线变离线；
用路径压缩的并查集能直接查询当前状态下某个人的最高boss；
签署一份文件时，记录签署此文件最底端的人和最顶端的人；
倍增处理出每个节点往上走1<<i步是谁；
处理出每个节点的深度deep[i]。

对于每个查询：
从第i个文件的最底端往上走(deep[x]-deep[i文件底端])步；
并查询走到的这个节点是否为x。

代码：

//jerry99
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int tmax=1e5+5;
int n,m,fa[tmax][20],f[tmax],deep[tmax];
int Start[tmax],End[tmax],num,numq;
struct node{
    int people,packnum;
};
node q[tmax];
vector<int> G[tmax];
int find(int x)
{
    return f[x]==x?x:f[x]=find(f[x]);
}
void dfs(int x,int d)
{
    deep[x]=d;
    int i,len=G[x].size();
    for(i=0;i<len;i++)
        dfs(G[x][i],d+1);
    return;
}
void init()
{
    int i,j;
    for(j=1;j<=17;j++)
        for(i=1;i<=n;i++)
            fa[i][j]=fa[fa[i][j-1]][j-1];
    for(i=1;i<=n;i++)
        if(find(i)==i)
            dfs(i,1);
    return;
}
void work()
{
    int i,j,Dif,s,t,check;
    for(j=1;j<=numq;j++)
    {
        s=Start[q[j].packnum];
        t=End[q[j].packnum];
        check=q[j].people;
        //printf("s=%d t=%d\n",s,t);
        if(find(s)!=find(check)||deep[s]<deep[check]||deep[check]<deep[t])
        {
            printf("NO\n");
            continue;
        }
        Dif=deep[s]-deep[check];
        for(i=17;i>=0;i--)
        {
            if(Dif<(1<<i)) continue;
            Dif-=(1<<i);
            s=fa[s][i];
            if(s==0) break;
        }
        if(s!=check) printf("NO\n");
        else printf("YES\n");
    }
    return;
}
int main()
{
    scanf("%d%d",&n,&m);
    int i,t,x,y;
    for(i=1;i<=n;i++)
        f[i]=i;
    for(i=1;i<=m;i++)
    {
        scanf("%d",&t);
        if(t==1)
        {
            scanf("%d%d",&x,&y);
            fa[x][0]=y;
            G[y].push_back(x);
            f[x]=find(y);
        }
        else if(t==2)
        {
            num++;
            scanf("%d",&Start[num]);
            End[num]=find(Start[num]);
        }
        else
        {
            numq++;
            scanf("%d%d",&q[numq].people,&q[numq].packnum);
        }
    }
    init();
    work();
    return 0;
}