HDU 3954 Level up（线段树）

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>

using namespace std;

const int maxn = 10010;
int t, n, k, qw;
int need[11] = {0};
int maxexp[maxn<<2][11];
int add[maxn<<2];

void build(int l, int r, int rt)
{
    for (int i = 1; i <= k; ++i)
        maxexp[rt][i] = -1;
    maxexp[rt][1] = 0;
    add[rt] = 0;
    if (l == r) return ;
    int m = (l + r) >> 1;
    build(l, m, rt << 1);
    build(m + 1, r, rt << 1 | 1);
}

void pushDown(int rt)
{
    if (add[rt])
    {
        add[rt<<1] += add[rt];
        add[rt<<1|1] += add[rt];
        for (int i = 1; i <= k; ++i)
        {
            if (maxexp[rt<<1][i] != -1)
                maxexp[rt<<1][i] += i * add[rt];
            if (maxexp[rt<<1|1][i] != -1)
                maxexp[rt<<1|1][i] += i * add[rt];
        }
        add[rt] = 0;
    }
}

void pushUp(int rt)
{
    for (int i = 1; i <= k; ++i)
    {
        maxexp[rt][i] = max(maxexp[rt<<1][i], maxexp[rt<<1|1][i]);
    }
}

void levelUp(int l, int r, int rt)
{
    if (l == r)
    {
        for (int i = 1; i < k; ++i)
        {
            if (maxexp[rt][i] != -1)
            {
                while (maxexp[rt][i] >= need[i+1] && i < k)
                {
                    maxexp[rt][i+1] = maxexp[rt][i];
                    maxexp[rt][i] = -1;
                    i++;
                }
                break;
            }
        }
        return ;
    }
    pushDown(rt);
    int m = (l + r) >> 1;
    for (int i = 1; i < k; ++i)
    {
        if (maxexp[rt<<1][i] >= need[i+1])
        {
            levelUp(l, m, rt << 1);
            break; 
        }
    }
    for (int i = 1; i < k; ++i)
    {
        if (maxexp[rt<<1|1][i] >= need[i+1])
        {
            levelUp(m + 1, r, rt << 1 | 1);
            break; 
        }
    }
    pushUp(rt);
}

void update(int l, int r, int L, int R, int e, int rt)
{
    if (L <= l && R >= r)
    {
        add[rt] += e;
        for (int i = 1; i <= k; ++i)
        {
            if (maxexp[rt][i] != -1)
            {
                maxexp[rt][i] += i * e;
            }
        }
        for (int i = 1; i < k; ++i)
        {
            if (maxexp[rt][i] >= need[i+1])
            {
                levelUp(l, r, rt);
                break; 
            }
        }
        return ;
    }
    if (l == r) return ;
    pushDown(rt);
    int m = (l + r) >> 1;
    if (L <= m) update(l, m, L, R, e, rt << 1);
    if (R > m) update(m + 1, r, L, R, e, rt << 1 | 1);
    pushUp(rt);
}

int query(int l, int r, int L, int R, int rt)
{
    if (L <= l && R >= r)
    {
        int ret = -1;
        for (int i = 1; i <= k; ++i)
        {
            ret = max(ret, maxexp[rt][i]);
        }
        return ret;
    }
    pushDown(rt);
    int m = (l + r) >> 1;
    int ret = -1;
    if (L <= m) ret = max(ret, query(l, m, L, R, rt << 1));
    if (R > m) ret = max(ret, query(m + 1, r, L, R, rt << 1 | 1));
    return ret;
}

int main()
{
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    scanf("%d", &t);
    for (int cas = 1; cas <= t; ++cas)
    {
        printf("Case %d:\n", cas);
        scanf("%d%d%d", &n, &k, &qw);
        for (int i = 2; i <= k; ++i)
        {
            scanf("%d", &need[i]);
        }
        char op[3];
        int li, ri, ei;
        build(1, n, 1);
        for (int i = 0; i < qw; ++i)
        {
            scanf("%s", op);
            if (op[0] == 'W')
            {
                scanf("%d%d%d", &li, &ri, &ei);
                update(1, n, li, ri, ei, 1);
            }
            else
            {
                scanf("%d%d", &li, &ri);
                printf("%d\n", query(1, n, li, ri, 1));
            }
        }
        printf("\n");
    }
    //system("pause");
    return 0;
}