洛谷 P2441 角色属性树

题目描述

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

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

举个例子，正妹是双份的猫耳，而且有一份弱气，她的属性值为2*2*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。

输入输出样例

输入样例：

4 6
10 8 4 3
1 2
2 3
3 4
1 1
1 2

 #include <bits/stdc++.h>

 using namespace std;
 #define inf 0x3f3f3f3f
 #define M 200200
 int next[M],pre[M],last[M],n,query,meng[M],tot=,father[M];
 bool vis[M];
 int gcd (int a,int b){return b==?a:gcd(b,a%b);}//1行代码gcd
 void addedge (int u,int v)
 {
     tot++;
     next[tot]=v;//next数组存的是点
     pre[tot]=last[u];//pre和last存的是边的编号
     last[u]=tot;
 }
 void dfs (int x)
 {
     for (int i=last[x];i;i=pre[i])
 father[next[i]]=x,dfs(next[i]);//每个点找上司
 //因为树是按顺序建的，所以直接无脑搜索就行了，不用vis，因为不可能走重
 }
 int main()

 {
     //freopen("de.txt","r",stdin);
     scanf("%d%d",&n,&query);
     memset(vis,false,sizeof vis);
     for (int i=;i<=n;++i)
     scanf("%d",&meng[i]);
     for (int i=;i<n;++i)
     {
         int x,y;
         scanf("%d%d",&x,&y);
         addedge(x,y);
     }
     dfs();
     for (int i=;i<query;++i)
     {
         int x,y,z,ans=-;
         scanf("%d%d",&x,&y);
         if (x==)
         scanf("%d",&z),meng[y]=z;
         else
         {
             for (int j=father[y];j;j=father[j])
             {
                 int a=meng[y],b=meng[j];
                 if (a>b) swap(a,b);
                 if (gcd(a,b)!=)
                 {
                     ans=j;
                     break;
                 }
             }
             printf("%d\n",ans);
         }
     }
     return ;
 }

1 3
1 4
2 1 9
1 4

输出样例：

-1
1
2
-1
1

这道题看评论很水...反正就是练一下树状的结构。顺便学会了1行gcd。
代码如下：