CF_17C Balance

C. Balance

http://codeforces.com/problemset/problem/17/C

题意：

  给出一个最多150字符长的只有a、b、c组成的字符串，每次操作可以选择两个连续的字符：
   1.把前面一个字符变成后面一个字符 。
   2.把后面一个字符变成前面一个字符 。
  (即可以 str[i+1] = str[i] 或者 str[i] = str[i+1]。)
  如果原字符串在执行若干次操作后变成一个a,b,c的字符数量差相互不超过1的 字符串, 那么称得到的串为一个合法串。
  求有多少个合法串。（ 取模 51123987）

数据：

n (1 ≤ n ≤ 150)

思路：

  1. 一段连续的字母必定对应原串的某个字符， 且原串的某个字符也必定对应可以得到的串中一段连续的字母（可为空）。 
    则可以去除原串中连续重复字符。只保留字符间的相对关系。
  2. dp[p][i][j][k] 表示当前已经有i个字母’a’,j个字母’b’,k个字母’c’,现在匹配的原串字符是第l个时，不同的子串数目。
  3. next[p][ch] 表示原串在第p个字符之后（包含第p个字符）第一次出现字符ch的位置。
  4. 转移方程：

dp[nxt[p][0]][i+1][j][k] += dp[p][i][j][k]
dp[nxt[p][1]][i][j+1][k] += dp[p][i][j][k] 
dp[nxt[p][2]][i][j][k+1] += dp[p][i][j][k] 

5. 当  i+j+k = n  且  i,j,k 满足相互差值不大于1 时，即累加答案。
6. 因为a,b,c最大都只可能有 n/3+ 1，所以实际复杂度是O( n^4 / 27 ) .

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <math.h>
#include <algorithm>
using namespace