牛客网WY-29 暗黑子字符串

牛客网WY-29 暗黑子字符串

题目大意：

一个只包含’A’、‘B’和’C’的字符串，如果存在某一段长度为3的连续子串中恰好’A’、‘B’和’C’各有一个，那么这个字符串就是纯净的，否则这个字符串就是暗黑的。例如：
BAACAACCBAAA 连续子串"CBA"中包含了’A’,‘B’,‘C’各一个，所以是纯净的字符串 AABBCCAABB
不存在一个长度为3的连续子串包含’A’,‘B’,‘C’,所以是暗黑的字符串
你的任务就是计算出长度为n的字符串(只包含’A’、‘B’和’C’)，有多少个是暗黑的字符串。

输入描述:

输入一个整数n，表示字符串长度(1 ≤ n ≤ 30)

输出描述:

输出一个整数表示有多少个暗黑字符串

示例1
输入

2 3

输出

9 21

解题思路：
本题要求求出不存在连续的三个为’A’ ,‘B’ ,‘C’的子串的个数，那么我们就要考虑任一长度下的字符与前两个字符之间的关系。再由这个关系从小到大推出所有长度的字符串存在的暗黑字符串。关系就是三个字母不能同时存在，那么我们可以分为两种情况：
第一种：前两个字符相同，我们设第i个字符在这个情况下存在的字符串数为same[i]；
第二种：前两个字符不同，我们设第i个字符在这个情况下存在的字符串数为different[i]；
由此得出关系式：
(1) dp[i]=same[i]+different[i];
(dp[i]表示为第i个字符下存在暗黑字符串的个数）
第一种情况下，前两个字符相同，那么第三个字符就可以为’A’,‘B’,‘C’这三个，例如：‘AAA’,‘AAB’,‘AAC’，因此可以推出关系式：
(2) same[i]=3*same[i-1];
第二种情况下，前两个字符不同，那么第三个字符就必须和前两个字符中的一个相同，例如：’ABA’,‘ABB’，因此可以推出关系式：
(3) different[i]=2*different[i-1];
将(2)和(3)代入到(1)得：
(4) dp[i]=2*dp[i-1]+same[i-1];
此时式中还剩下same[i-1]没有消去那么我们就要去更前面的状态dp[i-2]找same[i-1]和dp[i-2]的关系。

再观察这两种情况：
第一种是在前两个字符相同的情况下的same[i-1]的3种情况，接上面的例子分析：‘AAA’,‘AAB’,‘AAC’，这三个当中只有’AAA’相同，属于same[i]，其他两个不同，属于different[i]。

第二种是在前两个字符不同的情况下different[i-1]的2种情况，同样接上面的例子：‘ABA’,'ABB’两个当中只有’ABB’相同，属于same[i]
因此综合以上两种情况：
(5) same[i]=same[i-1]+different[i-1]
(6) different[i]=2*same[i-1]+different[i-1]
由（5）推出
(7) same[i]=dp[i-1]
(8) same[i-1]=dp[i-2]
代入（8）到（4）得出最后的关系式：
dp[i]=2*dp[i-1]+dp[i-2];
题目说最多有30长度的字符串，那么我们就可以先打表把所有长度的dp都算出来，然后直接查询，就不用每次输入值都算一遍时间复杂度降到O(1)
以下是代码

#include <iostream>
using namespace std;
const int mx=35;
long long dp[mx];
int main()
{
    int n;
    dp[1]=3;
        dp[2]=9;
        for(int i=3;i<=30;i++)
        {
            dp[i]=2*dp[i-1]+dp[i-2];
        }
    while(cin>>n)
    {
        cout<<dp[n]<<endl;
    }
}