【AT5148】[AGC036E] ABC String（乱搞）

点此看题面

• 给定一个由"A","B","C"构成的字符串(s)，要求选出一个最长的子序列，使得三种字符出现次数相同且相邻字符不同。
• (|s|le10^6)

乱搞题

首先，连续一段相同的字符肯定不可能同时选择，我们可以先把它们压成一个。

假设此时出现次数由少到多分别是(A,B,C)。

我们想要让(C)的出现次数和(B)相同，第一种方式是直接删去，即把两端字符不相同的(C)全部删掉；第二种方式是和一个(A)一起删去，要求(AC)或(CA)的两端不能全为(B)（实际上没有这种情况，如果形成(BACB)，这个(C)在第一步中就会被删去）。

由于(C)的出现次数比(B)多，所以(C)不可能只与(B)相邻，相连的(AC)肯定存在，因此一定能通过这种方式使得(C)和(B)出现次数相同。

然后我们想要保持(B,C)数量之差不变，让它们的出现次数都变成(A)，只要每次删去一对(BC)，要求两端不全为(A)即可。

代码：(O(|s|))

#include<bits/stdc++.h>
#define Tp template<typename Ty>
#define Ts template<typename Ty,typename... Ar>
#define Reg register
#define RI Reg int
#define Con const
#define CI Con int&
#define I inline
#define W while
#define N 1000000
using namespace std;
int n,c[3];char s[N+5];
I void Del(char A,char B,char C)//删去相连的AB，使得B和C数量相同
{
	RI i=1,l=n;n=0;W(c[B-'A']^c[C-'A'])//B和C数量仍然不同
		(s[i]==A&&s[i+1]==B||s[i]==B&&s[i+1]==A)&&(s[n]^C||s[i+2]^C)?(--c[A-'A'],--c[B-'A'],i+=2):(s[++n]=s[i++]);//AB相连且两端不为C时才能删去
	W(i<=l) s[++n]=s[i++];s[n+1]=0;//把剩余部分存回来
}
int main()
{
	RI i,l;for(scanf("%s",s+1),l=strlen(s+1),i=1;i<=l;++i) s[n]^s[i]&&++c[(s[++n]=s[i])-'A'];s[n+1]=0;//连续字符压成一个
	char A=c[0]<=c[1]&&c[0]<=c[2]?'A':(c[1]<=c[2]?'B':'C'),C=c[2]>=c[0]&&c[2]>=c[1]?'C':(c[1]>=c[0]?'B':'C'),B='A'+'B'+'C'-A-C;//按出现次数排序
	for(l=n,n=0,i=1;i<=l&&c[C-'A']^c[B-'A'];++i) s[i]==C&&(!n||i==l||s[n]^s[i+1])?--c[C-'A']:s[++n]=s[i];//直接删去两端字符不同的C
	W(i<=l) s[++n]=s[i++];s[n+1]=0;return Del(A,C,B),Del(B,C,A),puts(s+1),0;//删去AC使BC数目相同，删去BC使ABC数目相同
}