[C++]牛客 WY29 暗黑的字符串

最新推荐文章于 2025-03-28 09:27:16 发布

原创最新推荐文章于 2025-03-28 09:27:16 发布 · 159 阅读

0 ·

CC 4.0 BY-SA版权

牛客刷题专栏收录该内容

22 篇文章

订阅专栏

该博客内容涉及一个编程问题，即计算长度为n且只包含'A'、'B'、'C'的字符串中，有多少个是暗黑的（没有长度为3的连续子串包含'A'、'B'、'C'）。博主给出了一段C++代码实现，通过动态规划求解此问题。代码首先初始化两个数组a和b，分别表示前一位置与当前位置相同和不同的字符串数量，然后通过迭代更新这两个数组的值。最后输出a[n]+b[n]作为暗黑字符串的数量。

描述
一个只包含’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

#include<iostream>

using namespace std;

int main() {
    int n=0;
    while (cin >> n){
        //a[]数组表示前一位(i-1)与本位(i)是相同的
        //b[]数组表示前一位(i-1)与后一位(i)是不同的
        //a[i] = a[i-1] + b[i-1];//从i-1位到i位，要使两位的字母相同，
        //那么i位只能与i-1位所有字母出现情况相同
        //b[i] = 2 * a[i-1] + b[i-1];//从i-1位到i位，要使两位的字母不同，
        //则对应a[i-1]只能选择其它两个字母；同时要使其为黑暗字符串，对于b[]而言，b[i]一定要与其b[i-2]的字母相同，因此b[i]的选择情况与b[i-1]一样
        long long a[31] = {0};
        long long b[31] = {0};
        a[1] = 3;
        b[1] = 0;
        for(int i = 2; i <= n; i++) {
            a[i] = a[i-1] + b[i-1];
            b[i] = 2*a[i-1] + b[i-1];
        }
        cout << a[n] + b[n] << " ";
    }
    return 0;
}