Salary Inequity

这里写图片描述
这里写图片描述

线段树
按照dfs序给每个员工进行标号,那么同一颗子树上的员工编号就是连续的数字,于是查询和修改都能转换成区间操作,套上线段树就好了。

#include <bits/stdc++.h>
using namespace std;

const int maxn = 1000010;
const int maxx = 0x3f3f3f3f;
struct Node {
    int x, y;
    Node() {}
    Node(int xx, int yy): x(xx), y(yy) {}
};
Node F[5*maxn];
int a[5*maxn];
int fa[maxn], range[maxn], num[maxn];
int x, y, n, m, tot;
vector<int> G[maxn];

void clean(int x, int l, int r) {
    F[x].x = a[x] = 0;
    F[x].y = maxx;
    if (l == r) return;
    int mid = (l+r)/2;
    clean(x*2, l, mid);
    clean(x*2+1, mid+1, r);
}

void init() {
    memset(fa, 0, sizeof(fa));
    for (int i = 1; i <= n; i++) G[i].clear();
    clean(1, 1, n);
    tot = 0;
}

void dfs(int x) {
    num[x] = ++tot;
    for (int i = 0; i < G[x].size(); i++) dfs(G[x][i]);
    range[x] = tot;
}

void check(int x) {
    F[x].x = max(F[x*2].x, F[x*2+1].x);
    F[x].y = min(F[x*2].y, F[x*2+1].y);
}

void insert(int x, int l, int r, int t, int v) {
    if (l == r) {
        F[x].x = F[x].y = v;
        return;
    }
    int mid = (l+r)/2;
    if (t <= mid) insert(x*2, l, mid, t, v);
    else insert(x*2+1, mid+1, r, t, v);
    check(x);
}

void pd(int x, int l, int r) {
    if (l == r) a[x] = 0;
    if (a[x] != 0) {
        F[x*2].x += a[x];
        F[x*2+1].x += a[x];
        F[x*2].y += a[x];
        F[x*2+1].y += a[x];
        a[x*2] += a[x];
        a[x*2+1] += a[x];
        a[x] = 0;
    }
}
void add(int x, int l, int r, int ll, int rr, int v) {
    pd(x, l, r);
    if (l == ll && r == rr) {
        a[x] += v;
        F[x].x += v, F[x].y += v;
        return;
    }
    int mid = (l+r)/2;
    if (rr <= mid) add(x*2, l, mid, ll, rr, v);
    else if (ll > mid) add(x*2+1, mid+1, r, ll, rr, v);
    else {
        add(x*2, l, mid, ll, mid, v);
        add(x*2+1, mid+1, r, mid+1, rr, v);
    }
    check(x);
}

Node find(int x, int l, int r, int ll, int rr) {
    pd(x, l, r);
    if (l == ll && r == rr) {
        return F[x];
    }
    int mid = (l+r)/2;
    if (rr <= mid) return find(x*2, l, mid, ll, rr);
    else if (ll > mid) return find(x*2+1, mid+1, r, ll, rr);
    else {
        Node temp1 = find(x*2, l, mid, ll, mid);
        Node temp2 = find(x*2+1, mid+1, r, mid+1, rr);
        temp1.x = max(temp1.x, temp2.x);
        temp1.y = min(temp1.y, temp2.y);
        return temp1;
    }
}

char getOP() {
    char op = ' ';
    while (op != 'Q' && op != 'R') op = getchar();
    return op;
}

int main() {
    freopen("input.txt","r",stdin);
    int T;
    scanf("%d", &T);
    while (T--) {
        scanf("%d", &n);
        init();
        for (int i = 2; i <= n; i++) {
            scanf("%d", &fa[i]);
            G[fa[i]].push_back(i);
        }
        dfs(1);
        for (int i = 1; i <= n; i++) {
            scanf("%d", &x);
            insert(1, 1, n, num[i], x);
        }
        scanf("%d", &m);
        while (m--) {
            char c = getOP();
            if  (c == 'Q') {
                scanf("%d", &x);
                Node temp = find(1, 1, n, num[x], range[x]);
                printf("%d\n", temp.x-temp.y);
            } else {
                scanf("%d %d", &x, &y);
                add(1, 1, n, num[x], range[x], y);
            }
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值