树形结构转线性结构,先dfs,得到每一个结点的开始和结束访问时间s,t
记录一个数组A[1...t]
那么更新一个结点的就是A[s]+=v,A[t]-=v
采用树状数组做,OMlog2*N复杂度
#include<iostream>
#include<cstdio>#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int N = 1001000;
int C[N];
int wage[N];
vector<int> edge[N];
int st[N],ed[N];
int t = 1;
void dfs(int u){
st[u] = t++;
for(int i = 0;i<edge[u].size();i++){
dfs(edge[u][i]);
}
ed[u] = t++;
}
int lowbit(int x)//计算lowbit
{
return x&(-x);
}
void add(int i,int val)//将第i个元素更改为val
{
while(i<t)
{
C[i]+=val;
i+=lowbit(i);
}
}
int sum(int i)//求前i项和
{
int s=0;
while(i>0)
{
s+=C[i];
i-=lowbit(i);
}
return s;
}
int main(){
int n,m;
scanf("%d%d",&n,&m);
scanf("%d",&wage[1]);
memset(C,0,sizeof(C));
int v;
for(int i = 2;i<=n;i++){
scanf("%d%d",&wage[i],&v);
edge[v].push_back(i);
}
dfs(1);
char str[10];
int u;
for(int i = 0;i<m;i++){
scanf("%s",str);
if(str[0]=='u'){
scanf("%d",&u);
int s = sum(st[u]-1);
printf("%d\n",wage[u]+s);
}else {
scanf("%d%d",&u,&v);
add(st[u],v);
add(ed[u],-v);
}
}
return 0;
}
本文介绍了一种将树形结构转化为线性结构的方法,通过深度优先搜索(DFS)获取节点的访问时间,并利用树状数组进行高效更新,实现O(logN)的时间复杂度。

被折叠的 条评论
为什么被折叠?



