2019中国大学生程序设计竞赛-女生专场（重现赛）-Tree

Tree

Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 524288/524288 K (Java/Others)
Total Submission(s): 185 Accepted Submission(s): 64

Problem Description
wls 有三棵树，树上每个节点都有一个值 ai，现在有 2 种操作：

1. 将一条链上的所有节点的值开根号向下取整；
2. 求一条链上值的和；
   链的定义是两点之间的最短路。

Input
第一行两个数 n, q 分别代表树上点的数量和操作数量。
第二行 n 个整数，第 i 个数代表第 i 个点的值 ai。
接下来 n − 1 行， 每行两个整数 u, v 代表 u，v 之间有一条边。数据保证点两两联通。
接下来 q 行，每行有个整数 op, u, v，op = 0 表示将 u, v 这条链上所有的点的值开根号向下取整，op = 1表示询问 u，v 这条链上的值的和。
1 ≤ n, q ≤ 100, 000
0 ≤ ai ≤ 1, 000, 000, 000

Output
对于每一组 op = 2 的询问，输出一行一个值表示答案。

Sample Input
4 4
2 3 4 5
1 2
2 3
2 4
0 3 4
0 1 3
1 2 3
1 1 4

Sample Output
2
4

破题：看到树不要慌 建树最重要做了两天树，要会写dfs（）；写出来以后这个树还不是任你操作了？？
不要怜惜那内存！！；

思路：
dfs建树
struct node只有一个元素但是是个模板可以多次用的！！；
反正就是把一个无向图从1这个节点提起来顺着给的数据往后处理的；
还有 函数 f（）；里面的要点是对节点处理不要忘记头尾和结束的点，可以输出看看有没有记录成功；
没有要求顺序之类的 就一下子压到栈里不漏就行；

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=100010;
struct node{
    int nt;
}; 
vector<node>v[maxn];
bool book[maxn];
int dth[maxn],pre[maxn],val[maxn];
int qwq[maxn];
void dfs(int x)
{
    book[x]=1;
    for(int i=0;i<v[x].size();i++)
    {
        if(book[v[x][i].nt])continue;
        pre[v[x][i].nt]=x;
        dth[v[x][i].nt]=dth[x]+1;
        dfs(v[x][i].nt);
    }
}
int f(int q,int x,int y){
    int cnt=0,flag=0;
    while(x!=y)
    {
        if(dth[x]>=dth[y]){
            qwq[cnt++]=x;
            x=pre[x];
            flag=0;
        }else{
            qwq[cnt++]=y;
            y=pre[y];
            flag=1;
        }
    }
    if(flag==0){
        qwq[cnt++]=x;
    }else{
        qwq[cnt++]=y;
    }
    
    if(q==0)
    {
        for(int i=0;i<cnt;i++){
            val[qwq[i]]=sqrt(val[qwq[i]]);
        }
    }else{    
        ll sum=0;
        for(int i=0;i<cnt;i++){
            //cout<<val[qwq[i]]<<endl;
            sum+=val[qwq[i]];
        } 
        cout<<sum<<endl;
    }
    return 0;
}
int main()
{
    int n,q;
        memset(book,0,sizeof(book));
        scanf("%d%d",&n,&q);
        for(int i=0;i<=n;i++)v[i].clear();
        int x,y;
        node a;
        for(int i=1;i<=n;i++)
        {
            cin>>val[i];
        }
        for(int i=0;i<n-1;i++)
        {
            scanf("%d%d",&x,&y);
            a.nt=y;
            v[x].push_back(a);
            a.nt=x;
            v[y].push_back(a);
        }
        dfs(1);
        for(int i=0;i<q;i++){
            int qu,a,b;
            cin>>qu>>a>>b;
            f(qu,a,b);
        }
    return 0;
}