1. 程式人生 > >ACM-ICPC 2018 瀋陽賽區網路預賽 J. Ka Chang (分塊+樹狀陣列+dfs序)

ACM-ICPC 2018 瀋陽賽區網路預賽 J. Ka Chang (分塊+樹狀陣列+dfs序)

題意
給你一顆樹,由兩種操作:
1.把這棵樹深度為 D 的點全部都加上一個值。
2.求以p為根節點的子樹的權值和是多少?
思路
對於樹上的東西,我們可以把他求一下DFS序,之後就可以把樹上的結構變成線性的結構,之後就是查詢和修改這兩個東西了。
關於修改:對於修改來說,當前層的點少的時候我們直接樹狀陣列暴力修改,對於當前層數的點太多的情況來說,我們把層數打上標記。
對於查詢,我們先查詢樹狀數組裡面的值,之後在查詢我們每一層中的標記,具體怎麼查詢,看程式碼把。
程式碼

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn = 1e5+10;

vector<int>V[maxn],D[maxn] , large;
int n,q , u,v , maxd;
int l[maxn] , r[maxn], de[maxn] , tar[maxn];
int cnt = 0;

struct BIT
{
    int tree[maxn];
    void init()
    {
        memset(tree,0,sizeof
(tree)); } void update(int pos,int val) { for(; pos <= n+1 ; pos += pos & -pos) tree[pos] += val; } long long query(int pos) { long long ans = 0; for(; pos > 0 ; pos -= pos & -pos) ans += tree[pos]; return ans; } }sum; void dfs(int
u,int d) { l[u] = ++cnt; maxd = max(maxd,d); D[d].push_back(l[u]); // 我們把相同層數的點放進去 for(auto & i : V[u]) { dfs(i,d+1); } r[u] = cnt; } signed main() { BIT sum; sum.init(); maxd = 0; scanf("%d%d",&n,&q); cnt = 0 ; for(int i = 0 ; i < n-1 ; i++) { scanf("%lld%lld",&u,&v); V[u].push_back(v); } dfs(1,0); int B = sqrt(n); for(int i = 0 ; i <= maxd ;i++) { if(D[i].size() > B) large.push_back(i); // 如果某一層中的點數大於我們的塊數就放進去 } while(q--) { int op;scanf("%lld",&op); if(op == 1) { int d,va;scanf("%lld%lld",&d,&va); if(D[d].size() <= B) { for(auto &i : D[d]) {sum.update(i,va);} } else tar[d] += va; } else { int p;scanf("%lld",&p); int L = l[p] , R = r[p]; long long ans = sum.query(R) - sum.query(L-1); for(auto &i : large) { ans += (upper_bound(D[i].begin(),D[i].end(), R) - lower_bound(D[i].begin() , D[i].end() , L)) *tar[i]; // 關於查詢,我們在每一層中查詢看 // 看我們這一層中有多少個包含這個區間的值,然後我們乘上標記就好了 } cout<<ans<<endl; } } }