特 别 说 明 特别说明 特别说明
这 道 题 老 师 只 讲 了 80 分 的 方 法 , 最 后 两 个 点 会 T L E , 请 大 佬 帮 忙 看 一 下 正 解 怎 么 做 。 这道题老师只讲了80分的方法,最后两个点会TLE,请大佬帮忙看一下正解怎么做。 这道题老师只讲了80分的方法,最后两个点会TLE,请大佬帮忙看一下正解怎么做。
树
输入
小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
样例说明
初始各个节点的权值依次为[1,2,1,1,2]。
第一个操作给节点2的权值增加3,会给节点2的儿子4、5的权值增加-3。此时各个节点的权值变成[1,5,1,-2,-1]。
第二个操作给节点1的权值增加2,会给节点1的儿子2、3的权值增加-2,然后会给节点2的儿子4、5的权值增加-(-2)。各个节点的权值变成[3,3,-1,0,1]。
数据范围
对于50%的数据,1≤n≤2000,1≤m≤2000。
对于100%的数据,1≤n≤100000,1≤m≤100000。
思路
这道题就是一道dfs,就按着他做。
若操作为2,就直接输出;不然,用dfs找儿子加权值(记得权值到这个点的儿子时要变相反数)。
代码
#include<cstdio>
using namespace std;
int n,m,x,y,z,a[100001],le[100001];
struct note
{
int to,next;
}e[100005];
void dfs(int now,int d)//dfs加权值
{
a[now]+=d;//加权值
for (int i=le[now];i;i=e[i].next) dfs(e[i].to,-d);//找这个点的儿子
}
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]=(note){y,le[x]}; le[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);//加权值
}
}