题目:
小L非常喜欢树。最近,他发现了一棵有趣的树。这棵树有n个节点(1到n编号),节点i有一个初始的权值ai。这棵树的根是节点1。
这棵树有一个特殊的性质:当你给节点i的权值加 val 的时候,节点i的所有儿子的权值都会加 -val。注意当你给节点i的儿子的权值加 -val 时,节点i的这个儿子的所有儿子的权值都会加 -(-val),以此类推。样例说明可以很好地帮助你理解这个性质。
有2种操作:
操作(a).“1 x val”表示给节点x的权值加val。
操作(b).“2 x”输出节点x当前的权值。
为了帮助小L更好地理解这棵树,你必须处理m个操作。
输入:
第一行包含2个整数n和m。
第二行包含n个整数a1,a2,…,an(1≤ai≤1000)。
接下来的n-1行,每行两个整数u和v(1≤u接下来的m行,每行包含2种操作的一种。每个操作都保证1≤x≤n,1≤val≤1000。
输出:
对于每个操作(b),输出一个整数,表示节点x当前的权值。
样例输入:
5 5
1 2 1 1 2
1 2
1 3
2 4
2 5
1 2 3
1 1 2
2 1
2 2
2 4
样例输出:
3
3
0
思路:
dfs搜每个儿子,然后权值(相反数)相加。
代码:
#include<iostream>
#include<cstdio>
using namespace std;
int a[100000],n,m,x,z,now[100000],y;
struct node
{
int w,p;
}e[1000005];
void dfs(int dep,int g)
{
a[dep]+=g;//权值相加
for(int i=now[dep];i;i=e[i].p)
dfs(e[i].w,-g);//dfs搜儿子
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
e[i]=node{y,now[x]};now[x]=i;//邻接表
}
for(int i=1;i<=m;i++)
{
scanf("%d%d",&x,&y);
if(x==2)//操作二
{
printf("%d\n",a[y]);
continue;
}
scanf("%d",&z);
dfs(y,z);//dfs
}
}