sicily 4876. PLAĆE 每周一赛二

来源:互联网 发布:淘宝哪家百乐 编辑:程序博客网 时间:2024/06/05 07:17

树形结构转线性结构,先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;
}
原创粉丝点击