HDU 4288 线段树 + 离线处理

传送门 ：HDU 4288

题解

先上代码， 候补

AC code:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;

#define mid ((l + r) >> 1)
#define ls rt << 1, l, mid
#define rs rt << 1 | 1, mid + 1, r

typedef long long LL;
const int N = 100000 + 5;
int a[N];

struct Node {
    int len;
    LL sum[5];
}node[N << 2];

struct ch {
    int v;//操作数
    int f;//f标志操作类型
};

void pushUp(int rt) {//核心算法, 考察线段树节点意义 ： 更新只考虑当前结点和子结点，看做一个小团体 
    node[rt].len = node[rt << 1].len + node[rt << 1 | 1].len;
    for (int i = 0; i < 5; ++i) {//把左孩子当做团体的左端直接给这边sum
        node[rt].sum[i] = node[rt << 1].sum[i];
    }
    for (int i = 0; i < 5; ++i) {//考虑左孩子长
        node[rt].sum[(i + node[rt << 1].len) % 5] += node[rt << 1 | 1].sum[i];
    }
}

void update(int rt, int l, int r, int u, int v) {

    if (l == r) {
        node[rt].sum[1] = v;
        node[rt].len += v ? 1 : -1;
        return;
    }

    if (u <= mid) update(ls, u, v);
    else update(rs, u, v);

    pushUp(rt);

}

int main() {

    //freopen("in.txt", "r", stdin);

    int n, v;
    char op[10];
    ch tmp;

    while (cin >> n) {
        vector<ch> opv;//存操作
        memset(node, 0, sizeof(node));
        int nn = 0;
        while (n--) {

            cin >> op;
            if (op[0] == 's')   tmp.f = 0;
            else {
                cin >> v;
                tmp.v = v;

                if (op[0] == 'a') {
                    tmp.f = 1;
                    a[++nn] = v;
                }
                else tmp.f = -1;
            }

            opv.push_back(tmp);
        }
        sort(a + 1, a + nn + 1);//排序离线预处理操作位置
        for (int i = 0; i < opv.size(); ++i) {
            if (!opv[i].f) cout << node[1].sum[3] << endl;
            else {
                int pos = upper_bound(a + 1, a + nn + 1, opv[i].v) - a - 1;//二分查找位置
                update(1, 1, nn, pos, opv[i].f == 1 ? opv[i].v : 0);//更新
            }
        }
    }

    return 0;
}