Codeforces 558E - A Simple Task

A Simple Task

题目链接

我们可以用区间覆盖的想法去解决这个问题。我们可以开 $26$ 颗线段树，去维护某个字母的区间覆盖。那么我们只要按照升序或者降序去遍历每个字母，首先求出某个字母在 $[L,R]$ 的个数 $cnt$ ，然后把原先的 $[L,R]$ 都覆盖为 $0$ ，因为已经放好前面的字母，所以也应该得知当前字母覆盖的起始位置 $pos$ ，那么我们将这个字母的 $[pos,pos+cnt-1]$ 覆盖为 $1$ ，再更新下一个字母的起始位置，也就是 $pos+=cnt$ ，就这样就等效完成了排序。最后遍历每颗线段树去填充区间即可得到答案。

#include <bits/stdc++.h>
#define lson rt<<1
#define rson (rt<<1)|1

using namespace std;

const int N = 1e5 + 10;

char str[N], ans[N];

int n, m, tree[30][N << 2], lz[30][N << 2];

void push_down(int p, int rt, int l, int r) {
	if (~lz[p][rt]) {
		int mid = l + r >> 1;
		lz[p][lson] = lz[p][rson] = lz[p][rt];
		tree[p][lson] = (mid - l + 1) * lz[p][rt];
		tree[p][rson] = (r - mid) * lz[p][rt];
		lz[p][rt] = -1;
	}	
}

void update(int p, int rt, int l, int r, int L, int R, int v) {
	if (L <= l && r <= R) {
		lz[p][rt] = v;
		tree[p][rt] = (r - l + 1) * v;
		return ;
	}
	push_down(p, rt, l, r);
	int mid = l + r >> 1;
	if (mid >= L) update(p, lson, l, mid, L, R, v);
	if (mid < R) update(p, rson, mid + 1, r, L, R, v);
	tree[p][rt] = tree[p][lson] + tree[p][rson];
}

int query(int p, int rt, int l, int r, int L, int R) {
	if (L <= l && r <= R) {
		return tree[p][rt];
	}
	push_down(p, rt, l, r);
	int mid = l + r >> 1, sum = 0;
	if (mid >= L) sum += query(p, lson, l, mid, L, R);
	if (mid < R) sum += query(p, rson, mid + 1, r, L, R);
	return sum;
}

void tu(int p, int rt, int l, int r) {
	if (tree[p][rt] == r - l + 1) {
		for (int i = l; i <= r; ++i) {
			ans[i] = (char)(p + 'a');
		}
		return ;
	}
	push_down(p, rt, l, r);
	int mid = l + r >> 1;
	if (tree[p][lson]) tu(p, lson, l, mid);
	if (tree[p][rson]) tu(p, rson, mid + 1, r);
}

int main() {
	memset(lz, -1, sizeof lz);
	scanf("%d%d%s", &n, &m, str + 1);
	for (int i = 1; i <= n; ++i) {
		update(str[i] - 'a', 1, 1, n, i, i, 1);
	}
	while(m--) {
		int L, R, op, s, e, add, pos;
		scanf("%d%d%d", &L, &R, &op);
		pos = L;
		if (op == 1) {
			s = 0; e = 26; add = 1;
		}
		else {
			s = 25; e = -1; add = -1; 
		}
		for (int i = s; i != e ; i += add) {
			int cnt = query(i, 1, 1, n, L, R);
			if (cnt) {
				update(i, 1, 1, n, L, R, 0);
				update(i, 1, 1, n, pos, pos + cnt - 1, 1);
				pos += cnt;
			}
		}
	}
	for (int i = 0; i < 26; ++i) {
		tu(i, 1, 1, n);
	}
	for (int i = 1; i <= n; ++i) {
		printf("%c", ans[i]);
	}
}