ARC113E Rvom and Rsrev

ARC113E Rvom and Rsrev

显然,最优是 b 尽量靠前且尽量多,而在满足 b 尽量多的时候,让 a 也尽量多。

考虑分类讨论。

aa 结尾

显然,不管怎样都不会删除 b,考虑怎样让 a 尽量多。

显然,最后的 b 全在开头且连在一起,所以删 a 的本质就是使 b 连起来。

那么我们一次删除 a 至少连接一次 b,至多连接两次 b

  • 选取一个 bAbb(a+)b 中的 a 进行删除能连接一次 b。(其中 A 表示两个及以上个 a 连起来,a+ 表示任意个 a 连起来)
  • 选取两个 bab 中的 a 进行删除能连接两次 b

所以先操作情况二,最后处理剩余 a ,这样能做到总的操作次数最少(不包括最后的 a 串)。

具体地,记录长度为 11a 串和长度大于 11aa 串的个数,分别记为 a1a1a2a2,不包括最后的 a 串。

最少 a 删除操作次数即为 a12+(a1mod2)+a2\lfloor\frac{a1}{2}\rfloor+(a_1 \bmod 2)+a2

bb 结尾

aa 有偶数个

显然,最后全删 a 的字典序最大。

aa 有奇数个

abab 结尾

奇数个 a​,显然 a 删不完,那么这个 ab 结尾的 b 肯定无法移动到前面。

显然,把 ab 前面的 a 全删完更优,前缀 b 的个数减一个连续。

abbabb 结尾

同上,把 abb 前面的 a 全删完更优,前缀 b 的个数减二个连续。

bbbbbb 结尾
(a+)(b+)

结果只有 a(b+)a(b+)

(a+)(…)(b+)

其中一定有 ba,一定要操作且仅操作一次 b

答案一定是 (b+)(a+) 其中 b 的个数少两个。

首先,操作 b 的时候一定选的是 bab,因为其余情况,都无法让结尾变成 a,而结尾不是 a 的情况得不到更长的 b 的前缀。

把它的 b 和末尾的 b 交换,一定能 b 少两个且以 a 结尾,将后面两个 b 提前。

交换完后又是以 a 结尾的情况,计算最后交换后长度为 11a 串和长度大于 11aa 串的变化量即可。


结束讨论,时间复杂度 O(S)\mathcal O(\sum|S|)

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
#define he putchar('\n')
#define ha putchar(' ')

inline int read() {
	int x = 0, f = 1;
	char c = getchar();
	while (c < '0' || c > '9') {
		if (c == '-') f = -1;
		c = getchar();
	}
	while (c >= '0' && c <= '9') {
		x = x * 10 + c - '0';
		c = getchar();
	}
	return x * f;
}

inline void write(int x) {
	if (x < 0) {
		putchar('-');
		x = -x;
	}
	if (x > 9) write(x / 10);
	putchar(x % 10 + '0');
}

const int _ = 2e5 + 10;

int T, n, cnt[2];

char s[_];

signed main() {
	T = read();
	while (T--) {
		scanf("%s", s);
		n = strlen(s);
		cnt[0] = cnt[1] = 0;
		for (int i = 0; i < n; ++i) cnt[s[i] - 'a']++;
		if (cnt[0] == 0 || cnt[1] == 0) {
			printf("%s\n", s);
			continue;
		}
		int a1 = 0, a2 = 0, nw = 0;
		for (int i = n - 1; i >= 0; --i)
			if (s[i] == 'b') {
				for (int j = i - 1; j >= 0; --j)
					if (s[j] == 'a') nw++;
					else if (nw) {
						if (nw == 1) a1++;
						else a2++;
						nw = 0;
					}
				if (nw) {
					if (nw == 1) a1++;
					else a2++;
					nw = 0;
				}
				break;
			}
		//		cout << a1 << " " << a2 << "!!!!\n";
		if (s[n - 1] == 'a') {
			for (int i = 1; i <= cnt[1]; ++i) printf("b");
			for (int i = 1; i <= cnt[0] - 2 * (a1 / 2 + a1 % 2 + a2); ++i) printf("a");
			printf("\n");
		} else if (cnt[0] % 2 == 0) {
			for (int i = 1; i <= cnt[1]; ++i) printf("b");
			printf("\n");
		} else if (s[n - 2] == 'a') {
			// ab
			for (int i = 1; i <= cnt[1] - 1; ++i) printf("b");
			printf("ab\n");
		} else if (s[n - 3] == 'a') {
			// abb
			for (int i = 1; i <= cnt[1] - 2; ++i) printf("b");
			printf("abb\n");
		} else {
			// bbb
			int la = 0, lb = n;
			for (int i = 0; i < n; ++i)
				if (s[i] == 'a') la = max(la, i);
				else lb = min(lb, i);
			if (la < lb) {
				// aaabbb
				printf("a");
				for (int i = 1; i <= cnt[1]; ++i) printf("b");
				printf("\n");
			} else {
				// aaa...bbb
				bool flg = 0;
				for (int i = 0; i <= n - 3; ++i)
					if (s[i] == 'b' && s[i + 1] == 'a' && s[i + 2] == 'a') flg = 1;
				if (flg) {
					// aaa...baa...bbb
					for (int i = 1; i <= cnt[1] - 2; ++i) printf("b");
					for (int i = 1; i <= cnt[0] - 2 * (a1 / 2 + a1 % 2 + a2 - 1); ++i) printf("a");
					printf("\n");
				} else {
					// aaa...ba...bbb
					for (int i = 1; i <= cnt[1] - 2; ++i) printf("b");
					for (int i = 1; i <= cnt[0] - 2 * ((a1 - 1) / 2 + (a1 - 1) % 2 + a2); ++i) printf("a");
					printf("\n");
				}
			}
		}
	}
	return 0;
}
posted @ 2022-07-26 15:37  蒟蒻orz  阅读(1)  评论(0编辑  收藏  举报  来源