1916: F
Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 178 Solved: 8
SubmitStatusWeb BoardDescription
给一颗树,有n个结点,编号为1到n,1为根节点,有两种操作,1 x y把x结点权值加y,2 x查询x到根节点所有结点的权值和.
每个结点权值初始化为0。
Input
第一行输入一个整数t,代表有t组测试数据。
每组数据第一行为两个整数n,m代表结点个数和操作次数。
接下来n-1行,每行两个整数a,b,表示a结点和b结点有一条边。
接下来m行每行一个操作格式如上。
0<=n<=50000 ,0<=m<=50000,0<=y<=50000
Output
对于每组查询输出一个整数表示结果
Sample Input
1
5 5
1 2
1 3
3 4
3 5
2 5
1 1 2
1 3 1
2 4
2 1Sample Output
0
3
2
超时无数次比赛结束后才只要要用到DFS序的知识对这个临时学 比葫芦画瓢
居然AC了 -_-|||
这是大神题解 好好参悟吧
维护一棵树的dfs序,然后用树状数组维护,更新点权的时候把dfs序中当前点第一个点更新正值,第二个点更新负值,查询时就是前缀和。
dfs序即由根深搜时,第一次遍历到x点时,序列中加入x点,退出x点时再加一个x点。
AC代码
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
#include <stack>
#include <vector>
using namespace std;
vector <int> edge[50010];
int id;
long long Sum[100010];
int in[50010];
int out[50010];
int cnt;
bool vis[50010];
void DFS(int u)
{
in[u] = ++cnt;//DFS序中的记录进入该节点的时间
int len = edge[u].size();
for(int i = 0; i < len; i++) {
int v = edge[u][i];
if(!vis[v]) {
vis[v]=true;
DFS(v);
}
}
out[u] = ++cnt;//记录出来该节点的时间
}
int Lowbit(int x){
return x&-x;
}
void Add(int x,int d,int n){
while(x<=n){
Sum[x]+=d;
x+=Lowbit(x);
}
}
long long sum(int x){
long long S=0;
while(x > 0)
{
S+=Sum[x];
x-=Lowbit( x);
}
return S;
}
void init(int n){
for(int i=0;i<=n;i++){
edge[i].clear();vis[i]=false;
}
cnt=0;
memset(Sum,0,sizeof(Sum));
}
int main(){
int T; scanf("%d",&T);
while(T--){
int n;
int m;
scanf("%d %d",&n,&m);
init(n);
for(int i=1;i<n;i++){
int a,b;
scanf("%d %d",&a,&b);
edge[a].push_back(b);
edge[b].push_back(a);
}
vis[1]=true;
DFS(1);
for(int i=1;i<=m;i++){
int a,b,c;
scanf("%d",&a);
if(a==1){//每次加入值的时候只影响了在他进入时间之后出来时间之前的节点
scanf("%d %d",&b,&c);
Add(in[b],c,cnt); //进入之后的权值全部加上
Add(out[b],-c,cnt);//出去之后权值再减去
}
else{
scanf("%d",&b);
printf("%lld\n",sum(in[b]));
}
}
}
return 0;
}