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;
}