Codeforces 588E. A Simple Task (线段树+计数排序思想)

题目链接：http://codeforces.com/contest/558/problem/E

题意：有一串字符串，有两个操作：1操作是将l到r的字符串升序排序，0操作是降序排序。

题解：建立26棵线段树，类似计数排序思想。

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 5;
struct SegTree {
    int lazy[27], sum[27], l, r;
}T[N << 2];
int a[27][N];

void pushup(int p, int c) {
    T[p].sum[c] = T[p << 1].sum[c] + T[(p << 1)|1].sum[c];
}

void pushdown(int p, int c) {
    if(T[p].lazy[c] != -1) {
        T[p << 1].sum[c] = T[p].lazy[c]*(T[p << 1].r - T[p << 1].l + 1);
        T[(p << 1)|1].sum[c] = T[p].lazy[c]*(T[(p << 1)|1].r - T[(p << 1)|1].l + 1);
        T[p << 1].lazy[c] = T[(p << 1)|1].lazy[c] = T[p].lazy[c];
        T[p].lazy[c] = -1;
    }
}

void build(int p, int c, int l, int r) {
    int mid = (l + r) >> 1;
    T[p].l = l, T[p].r = r, T[p].lazy[c] = -1;
    if(l == r) {
        T[p].sum[c] = a[c][l];
        return ;
    }
    build(p << 1, c, l, mid);
    build((p << 1)|1, c, mid + 1, r);
    pushup(p, c);
}

int query(int p, int c, int l, int r) {
    int mid = (T[p].l + T[p].r) >> 1;
    if(T[p].l == l && T[p].r == r) {
        return T[p].sum[c];
    }
    pushdown(p, c);
    if(r <= mid) {
        return query(p << 1, c, l, r);
    } else if(l > mid) {
        return query((p << 1)|1, c, l, r);
    } else {
        return query(p << 1, c, l, mid) + query((p << 1)|1, c, mid + 1, r);
    }
    pushup(p, c);
}

void update(int p, int c, int l, int r, int val) {
    int mid = (T[p].l + T[p].r) >> 1;
    if(T[p].l == l && T[p].r == r) {
        T[p].lazy[c] = val;
        T[p].sum[c] = val * (r - l + 1);
        return ;
    }
    pushdown(p, c);
    if(r <= mid) {
        update(p << 1, c, l, r, val);
    } else if(l > mid) {
        update((p << 1)|1, c, l, r, val);
    } else {
        update(p << 1, c, l, mid, val), update((p << 1)|1, c, mid + 1, r, val);
    }
    pushup(p, c);
}
char str[N];

int main()
{
    int n, m;
    scanf("%d %d", &n, &m);
    scanf("%s", str);
    for(int i = 0; i < n; ++i) {
        a[str[i] - 'a'][i + 1] = 1;
    }
    for(int i = 0; i < 26; ++i) {
        build(1, i, 1, n);
    }
    while(m--) {
        int l, r, c;
        scanf("%d %d %d", &l, &r, &c);
        if(c) {
            int x = l, y = r;
            for(int i = 0; i < 26; ++i) {
                if(x > y)
                    break;
                int num = query(1, i, l, r);
                if(!num)
                    continue;
                update(1, i, l, r, 0);
                update(1, i, x, x + num - 1, 1);
                x = x + num;
            }
        } else {
            int x = l, y = r;
            for(int i = 25; i >= 0; --i) {
               if(x > y)
                    break;
                int num = query(1, i, l, r);
                if(!num)
                    continue;
                update(1, i, l, r, 0);
                update(1, i, x, x + num - 1, 1);
                x = x + num;
            }
        }
    }
    for(int i = 1; i <= n; ++i) {
        for(int j = 0; j < 26; ++j) {
            if(query(1, j, i, i)) {
                putchar(char(j + 'a'));
                break;
            }
        }
    }
    putchar('
');
    return 0;
}