ACM-ICPC 2018 瀋陽賽區網路預賽 J. Ka Chang (分塊+樹狀陣列+dfs序)
阿新 • • 發佈:2018-11-04
題意
給你一顆樹,由兩種操作:
1.把這棵樹深度為
的點全部都加上一個值。
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;
}
}
}