[51nod] 1199 Money out of Thin Air #线段树+DFS序

1199 Money out of Thin Air

题目来源： Ural 基准时间限制：1 秒 空间限制：131072 KB 分值: 80 难度：5级算法题

 

一棵有N个节点的树，每个节点对应1个编号及1个权值，有2种不同的操作。

操作1：S x y z，表示如果编号为x的节点的权值 < y，则将节点x的权值加上z。(Single)

操作2：A x y z，表示如果编号为x的节点以及其所有子节点的权值平均值 < y，则将节点x及其所有子节点的权值加上z。(All)

给出树节点之间的关系，进行M次操作，问所有操作完成后，各个节点的权值为多少？

节点的编号为0 - N - 1，根节点的编号为0，并且初始情况下，根节点的权值也是0。

 

Input

第1行：2个数N, M，N为节点的数量，M为操作的数量(1 <= N, M <= 50000)。
第2 - N行：每行描述一个节点N[i]的信息，第2行对应编号为1的节点，第N行对应编号为N - 1的节点。具体内容为：每行2个数P[i], W[i]。P[i]为当前节点的父节点的编号，W[i]为当前节点的权值。(0 <= W[i] <= 10^5, P[i] < i)
第N + 1 - N + M行：每行表示一个操作，S x y z或A x y z，（0 <= y, z <= 10^5)。

Output

输出共N行，每行1个数W[i]，表示经过M次后，编号为0 - N - 1的节点的权值。

Input示例

4 3
0 10
0 10
1 2
S 0 1 1
A 0 20 1
S 3 2 1

Output示例

2
11
11
3

Analysis分析

 求树的DFS序并用线段树维护，经典题型

 

Code代码

#include<stdio.h>
#include<iostream>
#define mid (L+R)/2
#define lc (rt<<1)
#define rc (rt<<1|1)
#define maxn 1000000
#define LL long long
using namespace std;

int dfn[maxn],dfl[maxn],TIM,sz[maxn],fa[maxn],n,m;
LL vval[maxn],arr[maxn],ans[maxn];

struct edge{
    int from,v;
}e[maxn]; int tot,first[maxn];
void insert(int u,int v){ tot++; e[tot].from = first[u]; e[tot].v = v; first[u] = tot; }

void dfs(int u){
    dfn[u] = ++TIM;
    arr[TIM] = vval[u];
    sz[u] = 1;
    for(int i = first[u];i;i = e[i].from){
        int v = e[i].v;
        dfs(v);
        sz[u] += sz[v];
    }dfl[u] = TIM;
}

struct node{
    LL sum,lazy;
}T[maxn*4];

void pushdown(int rt,int L,int R){
    if(!T[rt].lazy) return;
    T[lc].lazy += T[rt].lazy;
    T[rc].lazy += T[rt].lazy;
    T[lc].sum += T[rt].lazy*(mid-L+1);
    T[rc].sum += T[rt].lazy*(R-mid);
    T[rt].lazy = 0;
}

void build(int rt,int L,int R){ // 1,1,TIM
    if(L == R) T[rt].sum = arr[L];
    else{
        build(lc,L,mid); build(rc,mid+1,R);
        T[rt].sum = T[lc].sum+T[rc].sum;
    }
}

LL query(int rt,int L,int R,int qL,int qR){
    pushdown(rt,L,R);
    if(qL <= L && R <= qR) return T[rt].sum;
    else{
        LL ret = 0;
        if(qL <= mid) ret += query(lc,L,mid,qL,qR);
        if(qR > mid) ret += query(rc,mid+1,R,qL,qR);
        return ret;
    }
}

void modify(int rt,int L,int R,int qL,int qR,LL val){
    pushdown(rt,L,R);
    if(qL <= L && R <= qR){
        T[rt].sum += val*(R-L+1);
        T[rt].lazy += val;
    }else{
        if(qL <= mid) modify(lc,L,mid,qL,qR,val);
        if(qR > mid) modify(rc,mid+1,R,qL,qR,val);
        T[rt].sum = T[lc].sum + T[rc].sum;
    }
}

void print(int rt,int L,int R){
    pushdown(rt,L,R);
    if(L == R) ans[L] = T[rt].sum;
    else{
        print(lc,L,mid);
        print(rc,mid+1,R);
    }
}

void final(){
    for(int i = 1;i <= n;i++) printf("%lld
",ans[dfn[i]]);
}

int main(){
//    freopen("1.in","r",stdin);
    
    scanf("%d%d",&n,&m);
    
    for(int i = 2;i <= n;i++){
        scanf("%d%lld",&fa[i],&vval[i]);
        insert(fa[i]+1,i);
//        insert(fa[i],i);
    }
    
    dfs(1);
    build(1,1,TIM);
    
    for(int i = 1;i <= m;i++){
        char ctr; int x; LL z; double y;
        cin >> ctr; scanf("%d%lf%lld",&x,&y,&z);
        
        x++;
        if(ctr == 'A'){
            LL sum = query(1,1,TIM,dfn[x],dfl[x]);
            if(1.0*sum/(dfl[x]-dfn[x]+1) < y) modify(1,1,TIM,dfn[x],dfl[x],z);
        }else if(1.0*query(1,1,TIM,dfn[x],dfn[x]) < y) modify(1,1,TIM,dfn[x],dfn[x],z);
//        cout << "BBB" << endl;
    }
    
    print(1,1,TIM);
    final();
    
    return 0;
}