CF17C Balance

题意

【题目描述】 一个仅由a,b,c三种字符组成的字符串，可以对其进行如下两种操作：

选择两个相邻字符，将第一个字符替换成第二个。

选择两个相邻字符，将第二个字符替换成第一个。 这样，通过任意多次的操作，可以得到许多不同的串，为了追求字符的平衡， 我们要求a,b,c三种字符在字符串中出现的次数两两之差都不能大于1。

你的任 务是，统计给定字符串通过任意多次的操作，能够得到的不同的平衡串的个数。

【输入格式】

输入文件包含2行。 第一行包含1个正整数n，表示字符串长度。 第二行包含1个长度为n的字符串。

【输出格式】

输出共1行，包含1个正整数，表示能够得到的平衡串的数量。由于答案可 能很大，因此输出时要对51123987取模。

输入输出样例

输入样例#1：

4

abca

输出样例#1：

7

输入样例#2：

4

abbc

输出样例#2：

3

输入样例#3：

2

ab

输出样例#3：

1

---

f[i][j][k][l]表示以第i个点开始向右扩展, 状态有j个a,k个b,l个c的方案数

因为扩展出来的序列是连续的一段

所以可以把序列中一段相同的数给看成是一个数

从这个点开始可以一直扩展到头

所以实时统计当前状态是否是合法的

如果是合法的就统计到答案里

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
const int M = 155 ;
const int N = 55 ;
const int mod = 51123987 ;
using namespace std ;
int n ;
char s[M] ;
int v[M] , Nxt[M][3] , upp , Ans , f[M][N][N][N] ;
int main() {
	scanf("%d",&n) ;
	scanf("%s",s + 1) ;
	for(int i = 1 ; i <= n ; i ++) v[i] = s[i] - 'a' ;
	Nxt[n + 1][0] = Nxt[n + 1][1] = Nxt[n + 1][2] = n + 1 ;
	for(int i = n ; i >= 1 ; i --) {
		Nxt[i][0] = Nxt[i + 1][0] ; Nxt[i][1] = Nxt[i + 1][1] ; Nxt[i][2] = Nxt[i + 1][2] ;
	    if(v[i] == 0) Nxt[i][0] = i ;
	    if(v[i] == 1) Nxt[i][1] = i ;
	    if(v[i] == 2) Nxt[i][2] = i ;
	}
	upp = n / 3 + 2 ;
	f[1][0][0][0] = 1 ;
	for(int i = 1 ; i <= n ; i ++)
	    for(int j = 0 ; j <= upp ; j ++)
	        for(int k = 0 ; k <= upp ; k ++)
	            for(int l = 0 ; l <= upp ; l ++) {
	            	if(j + k + l == n && abs(j - k) <= 1
					&& abs(j - l) <= 1 && abs(k - l) <= 1)
					    Ans = (Ans + f[i][j][k][l]) % mod ;
					f[Nxt[i][0]][j + 1][k][l] = (f[Nxt[i][0]][j + 1][k][l] + f[i][j][k][l]) % mod ;
					f[Nxt[i][1]][j][k + 1][l] = (f[Nxt[i][1]][j][k + 1][l] + f[i][j][k][l]) % mod ;
					f[Nxt[i][2]][j][k][l + 1] = (f[Nxt[i][2]][j][k][l + 1] + f[i][j][k][l]) % mod ;
				}
	cout << Ans << endl ;
	return 0 ;
}