CF558E-A Simple Task-线段树+计数排序

计数排序的原理，只要知道了有几个数比i小，就可以知道i的位置

这道题只有26个字母，搞26颗线段树，然后区间更新

#include <cstdio>
#include <cstring>
#include <algorithm>

//using namespace std;
const int maxn = 1e5+10;

int N,Q;
char line[maxn];

#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
#define root 1,1,N

struct SegrTree{
    int num[maxn<<2];
    int lazy[maxn<<2];
    void init()
    {
        memset(lazy,-1,sizeof lazy);
    }
    void push_up(int rt)
    {
        num[rt] = num[rt<<1] + num[rt<<1|1];
    }
    void push_down(int rt,int m)
    {
        if(lazy[rt] != -1)
        {
            num[rt<<1] = (m-(m>>1))*lazy[rt] ;
            num[rt<<1|1] = (m>>1)*lazy[rt] ;
            lazy[rt<<1] = lazy[rt<<1|1] = lazy[rt];
            lazy[rt] = -1;
        }
    }
    void build(int kind,int rt,int l,int r)
    {
        if(l == r)
        {
            num[rt] = line[l-1]==kind+'a';
            //if(line[l-1]=='c') printf("%c pos:%d 
",kind+'a',l);
            return ;
        }
        int mid = (l+r)>>1;
        build(kind,lson);
        build(kind,rson);
        push_up(rt);
    }
    void update(int L,int R,int d,int rt,int l,int r)
    {
        if(L <= l && R >= r)
        {
            num[rt] = (r-l+1)*d;
            lazy[rt] = d;
            return ;
        }
        push_down(rt,r-l+1);
        int mid = (l+r)>>1;
        if(L <= mid) update(L,R,d,lson);
        if(R >  mid) update(L,R,d,rson);
        push_up(rt);
    }
    int query(int L,int R,int rt,int l,int r)
    {
        if(L <= l && R >= r)
        {
            return num[rt];
        }
        push_down(rt,r-l+1);
        int mid = (l+r)>>1;
        int res = 0;

        if(R <= mid) res = query(L,R,lson);
        else if(L > mid) res = query(L,R,rson);
        else res = query(L,R,lson)+query(L,R,rson);

        push_up(rt);
        return res;
    }
}alpha[26];

void sort(int l,int r,int k)
{
    int pos,cnt;
    if(k == 1)
    {
        pos = l;
        for(int i=0;i<26;i++)
        {
            cnt = alpha[i].query(l,r,root);
            //printf("[%d,%d]%c k:%d pos:%d cnt:%d
",l,r,i+'a',k,pos,cnt);
            if(cnt)
            {
                alpha[i].update(l,r,0,root);
                alpha[i].update(pos,pos+cnt-1,1,root);
                pos += cnt;
            }
        }
    }
    else
    {
        pos = l;
        for(int i=25;i>=0;i--)
        {
            cnt = alpha[i].query(l,r,root);
            //printf("[%d,%d] %c k:%d pos:%d cnt:%d
",l,r,i+'a',k,pos,cnt);
            if(cnt)
            {
                alpha[i].update(l,r,0,root);
                alpha[i].update(pos,pos+cnt-1,1,root);
                pos += cnt;
            }
        }
    }
}

int main()
{
    //freopen("input.txt","r",stdin);

    scanf("%d%d",&N,&Q);
    scanf("%s",line);

    for(int i=0;i<26;i++)
    {
        alpha[i].init();
        alpha[i].build(i,root);
        //printf("tot num:%c %d
",i+'a',alpha[i].query(1,N,root));
    }
    //printf("c:%d c:%d
",alpha[2].query(1,N,root),alpha[2].query(7,10,root));
    for(int i=0,l,r,k;i<Q;i++)
    {
        scanf("%d%d%d",&l,&r,&k);
        sort(l,r,k);
    }
    for(int i=1;i<=N;i++)
    {
        for(int j=0;j<26;j++) if(alpha[j].query(i,i,root))
        {
            printf("%c",'a'+j);
        }
    }
    puts("");
}