树形数据结构——角色属性树

green qwq

于 2024-04-27 22:09:24 发布

阅读量793

点赞数 20

文章标签：数据结构深度优先算法

本文链接：https://blog.csdn.net/m0_60738889/article/details/138259357

版权

本文描述了一个基于角色属性树的组织结构问题，其中成员的萌点属性由质数组成，成员关心找到与其有相同萌元素的最近上司。文章涉及计算最大公约数（gcd）和最近公共祖先(lca)的方法来解决查询和属性修改问题。

摘要由CSDN通过智能技术生成

角色属性树

题目描述

绪萌同人社是一个有趣的组织，该组织结构是一个树形结构。有一个社长，直接下属一些副社长。每个副社长又直接下属一些部长……。

每个成员都有一个萌点的属性，萌点属性是由一些质数的萌元素乘积构成（例如，猫耳的值是 $2$ ，弱气的值是 $3$ ，黄毛的值是 $5$ ，病娇的值是 $7$ ，双马尾的值是 $11$ 等等）

举个例子，正妹是双份的猫耳，而且有一份弱气，她的属性值为 $2\times 2\times 3=12$ 。

现在组员关心一个问题，希望知道离自己最近且有相同萌元素上司是谁，例如，属性值为 $2 、 4 、 6 、 45$ 这样的属性值都算是和正妹有相同的属性。

然而，组员可能会随时变化自己的属性。啊。。感觉好麻烦啊。。

输入格式

第一行， $n, k$ 表示成员数与询问的次数

第二行， $n$ 个数，分别是 $1$ 到 $n$ 号成员的属性值

接下来 $n - 1$ 行， $x_i,y_i$ 表示 $x_i$ 是 $y_i$ 的上司。

接下来来 $k$ 行，有两种情况

$1\ u_i$ ：询问离 $u_i$ 成员最近且有相同萌元素上司。

$2\ u_i$ ： $a$ 更改 $u_i$ 的属性值为 $a$ 。

输出格式

对于每个 $1$ 类型的询问，输出符合要求的编号。如果没有符合要求的编号，输出 $- 1$ 。

样例 #1

样例输入 #1

样例输出 #1

提示

对于 $20\%$ 的数据，没有修改的操作。

对于 $50\%$ 的数据， $n\le 100$ ，修改次数 $< 10$ 。

对于 $100\%$ 的数据， $n\le 200000$ ， $k < 100000$ ，修改次数 $\le 50,a\_i\le 2^{31}-1$

UPD：本题测试数据随机，可能是假题。

思路

本道题我们直接用fa来装u的父节点，对于最近且有相同萌元素上司，我们可以当两者的gcd>1，那么就说明有最近且有相同萌元素。（为什么呢？依据属性值为 $2 、 4 、 6 、 45$ 这样的属性值都算是和正妹有相同的属性。和正妹是双份的猫耳，而且有一份弱气，她的属性值为 $2\times 2\times 3=12$ ）

代码

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int N = 200010,M = 2*N;

int e[M],ne[M],w[M],h[N],idx;
int fa[N];
bool st[N];
int n,k;

void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

//相同相同萌元素上司就是gcd>1的。

int gcd(int a,int b){
    return b?gcd(b,a%b):a;
}

int lca(int x,int y){
    if(!x)return -1;
    if(gcd(w[x],w[y])>1)return x;
    return lca(fa[x],y);
}

void dfs(int u,int f){
    fa[u]=f;
    for(int i=h[u];~i;i=ne[i]){
        int j=e[i];
        if(j==f)continue;
        dfs(j,u);
    }
}

int main(){
    cin>>n>>k;
    
    memset(h,-1,sizeof h);
    
    for(int i=1;i<=n;i++){
        cin>>w[i];    
    }
    
    for(int i=1;i<=n-1;i++){
        int a,b;
        cin>>a>>b;
        // fa[b]=a;//可以这么写
        add(a,b);
    }
    
    dfs(1,-1);//也可以这么写（我怎么感觉多此一举呢）

    
    for(int i=1;i<=k;i++){
        int op;
        cin>>op;
        if(op==1){
            int a;
            cin>>a;
            cout<<lca(fa[a],a)<<endl;
        }else{
            int a,b;
            cin>>a>>b;
            w[a]=b;
        }
    }
    
    return 0;
}