题意
给你一颗树,由两种操作:
1.把这棵树深度为
D
D
<script type="math/tex" id="MathJax-Element-3">D</script>的点全部都加上一个值。
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;
}
}
}