zzuli 1919 F

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;
}
發佈了95 篇原創文章 · 獲贊 0 · 訪問量 2萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章