百度之星2014复赛 - 1002 - The Query on the Tree

先上题目：

The Query on the Tree

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 54    Accepted Submission(s): 18

Problem Description

　　度度熊最近沉迷在和树有关的游戏了，他一直认为树是最神奇的数据结构。一天他遇到这样一个问题：
　　有一棵树，树的每个点有点权，每次有三种操作：
　　1. Query x 表示查询以x为根的子树的权值和。
　　2. Change x y 表示把x点的权值改为y(0<=y<=100)。
　　3. Root x 表示把x变为根。
　　现在度度熊想请更聪明的你帮助解决这个问题。

 

Input

　　第一行为数据组数T(1 <= T <= 100)
　　每组数据第一行为N(1<= N <= 10000)，表示树的节点数。
　　后面N-1行每行有两个数x,y ，表示x,y之间有一条边 1<=x,y<=N。初始时树是以1号节点为根节点。
　　之后的一行为N个数表示这N个点的点权(点权的范围是0到100)。
　　然后为整数Q（Q<=1000）为操作次数。
　　之后的Q行为描述中的三种操作。

 

Output

　　对于第k组输入数据，第一行输出Case #k 接下来对于每个”Query x”操作，输出以x为根的子数和。

 

Sample Input

2

5

1 2

1 3

3 4

3 5

1 2 3 4 5
5

Query 1

Change 3 10

Query 1

Root 4

Query 3

8

1 2

1 3

3 4

4 5

5 6

5 7

4 8

1 2 3 4 5 6 7 8

5

Query 1

Query 3

Root 5

Query 3

Query 1

 

Sample Output

Case #1:

15

22

18

Case #2:

36

33

6

3

 

Source

2014年百度之星程序设计大赛 - 复赛

 

　　中文题意不解释，思路记录每一个节点的当前父节点(根节点的父节点可记为0) ，然后对于三种操作，如果是修改某个结点的值，那就修改完这个值以及当前节点的子树的值以后，向父节点方向修改每一个节点的子树的值；如果是修改根节点，那就从新根节点开始向父节点方向移动，将遇到的节点的父节点都改成向新根节点方向，同时更新子树的最值；如果是查询的话，就直接输出。需要注意的是既的更新父节点以及子树的权值。

上代码：

 #include <cstdio>
 #include <cstring>
 #include <utility>
 #define MAX 10002
 using namespace std;
 typedef pair<int,int> pii;

 int n,tot;
 int p[MAX];
 int fa[MAX];
 int w[MAX];
 int s[MAX];
 pii e[MAX<<];
 char q[];
 bool f;

 void reset(){
     memset(p,-,sizeof(int)*(n+));
     tot=;
 }

 void add(int u,int v){
     e[tot].first=v; e[tot].second=p[u]; p[u]=tot++;
     e[tot].first=u; e[tot].second=p[v]; p[v]=tot++;
 }

 int deal(int u,int t){
     int c=;
     s[u]=;
     fa[u]=t;
     for(int v=p[u];v!=-;v=e[v].second){
         if(e[v].first!=t){
             c = deal(e[v].first,u);
             s[u]+=c;
         }
     }
     s[u]+=w[u];
     return s[u];
 }

 void C(int a,int v){
     int c = w[a];
     while(a!=){
         s[a] = s[a] - c + v;
         a = fa[a];
     }
 }

 void R(int r,int t){
     s[r]=;
     for(int i=p[r];i!=-;i=e[i].second){
         if(e[i].first==t) continue;
         if(e[i].first==fa[r]) R(fa[r],r);
         s[r]+=s[e[i].first];
         fa[e[i].first]=r;
     }
     s[r]+=w[r];
 }

 int main()
 {
     int t,u,v,a,m;
     //freopen("data.txt","r",stdin);
     scanf("%d",&t);
     for(int z=;z<=t;z++){
         printf("Case #%d:\n",z);
         scanf("%d",&n);
         reset();
         for(int i=;i<n;i++){
             scanf("%d %d",&u,&v);
             add(u,v);
         }
         for(int i=;i<=n;i++) scanf("%d",&w[i]);
         deal(,);
         scanf("%d",&a);
         while(a--){
             scanf("%s %d",q,&m);
             if(q[]=='Q'){
                 printf("%d\n",s[m]);
             }else if(q[]=='C'){
                 scanf("%d",&v);
                 C(m,v);
                 w[m]=v;
             }else{
                 R(m,);
                 fa[m]=;
             }
         }
     }
     return ;
 }

1002