codeforces466E 并查集+离线处理+dfs

来源：互联网发布：淘宝上怎么搜高仿手表编辑：程序博客网时间：2024/04/27 17:23

题目大意：一开始有n个员工，它们互相独立，现有三种操作：

1 x y y称为x的上级

2 x 从x发起一份文件，依次向上级传阅。文件编号依次递增

3 x y 问x是否查阅过编号为y的文件

思路：最暴力的解法就是模拟他的规则，但是这样明显超时。此时，想到了离线操作。我们可以用并查集确定每份文件的发起者和终结者，即文件移动的范围。。然后对于每次询问可以拆成两个标记，假设查询x是否浏览过第k号文件，第k号文件的范围为u-v，那么在最后dfs时，遍历到x，判断是否经过u；遍历到v时，判断是否经过x。如果两个都满足，则是YES。（此处参考他人做法）

#include <iostream>#include <cstdio>#include <string>#include <cstring>#include <fstream>#include <algorithm>#include <cmath>#include <queue>#include <stack>#include <vector>#include <map>#include <set>#include <iomanip>using namespace std;//#pragma comment(linker, "/STACK:102400000,102400000")#define maxn 200005#define MOD 1000000007#define mem(a , b) memset(a , b , sizeof(a))#define LL long long#define ULL unsigned long longtypedef pair<int , int> pii;const long long INF=0x3fffffff;int n , m;int fa[maxn] , vis[maxn] , ans[maxn];vector<int>v[maxn];vector<pii>q[maxn];struct node{    int st , ed;}rang[maxn] , tmp;int findfa(int a ){    if(fa[a] == a) return a;    else return fa[a] = findfa(fa[a] );}void dfs(int u ){    vis[u] = 1;    for (int i = 0; i < v[u].size(); i++)        dfs(v[u][i]);    for (int i = 0; i < q[u].size(); i++)     {        int v = q[u][i].first, id = q[u][i].second;        if (vis[v])            ans[id]++;    }    vis[u] = 0;}int main(){    while(scanf("%d %d" , &n , &m) != EOF)    {        mem(ans , 0 ) ;        int op , x , y , pos = 1 , K = 1;        for(int i = 0 ; i <= n ; i ++) fa[i] = i , v[i].clear();        for(int i = 1 ; i <= m ; i ++)        {            scanf("%d" , &op);            if(op == 1)            {                scanf("%d %d" , &x , &y);                fa[x] = y;                v[y].push_back(x);            }            else if(op == 2)            {                scanf("%d" , &x);                rang[pos].st = x;                rang[pos++].ed = findfa(x );            }            else            {                scanf("%d %d" , &x , &y);                pii u = make_pair(rang[y].ed , rang[y].st);                q[x].push_back(make_pair(u.first, K));    //将K号文件终止节点加入x ，当dfs到x时，判断是否经过文件的终止点                q[u.second].push_back(make_pair(x, K));   // 将查询的x放入K号文件的起点，dfs到K的起点时，判断是否经过x                K++;            }        }        for(int i = 1 ; i <= n ;i ++)        {            if(fa[i] == i)                dfs(i);        }        for(int i = 1 ; i < K ; i ++)        {            if(ans[i] == 2) printf("YES\n");            else printf("NO\n");        }    }    return 0;}

0 0