sicily 4876. PLAĆE 每周一赛二

本文介绍了一种将树形结构转化为线性结构的方法,通过深度优先搜索(DFS)获取节点的访问时间,并利用树状数组进行高效更新,实现O(logN)的时间复杂度。

树形结构转线性结构,先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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值