牛客网在线编程(19):暗黑字符串-python

版权声明:本文为博主原创文章,未经博主允许不得转载。转载请注明作者和出处:https://blog.csdn.net/weixin_41481113 https://blog.csdn.net/weixin_41481113/article/details/83956343

题目描述

一个只包含'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

代码实现

n=int(input())
mem=[3,9]
for i in range(n-2):
    mem=[mem[1],2*mem[1]+mem[0]]
print(str(mem[1]))

 原理说明

1.初始化一个数组mem,mem里初始存储的是n=1和n=2时的暗黑字符串数,当n>=3时,我们更新数组;

2.更新规则:f(n)=2*f(n-1)+f(n-2)      f(n)表示输入的整数为n时的暗黑字符串数;

                     在计算f(n)时,我们假设已知f(n-1),那么我们只需要在长度为n-1的暗黑字符串后面加上满足条件的‘A’,‘B’,                       ‘C’即可,假如某长度为n-1的暗黑字符串的最后两个字符不相同(比如‘AB’),那么我们在最后添加的字符只要不                         是这两个字符之外的第三个字符(‘C’)就可以得到长度为n的暗黑字符串,这时候我们有结论1:1个长度为n-1的                         暗黑字符串可以得到2个长度为n的暗黑字符串;那么如果最后两个字符相同(比如‘AA’)呢?那就更简单了,随便加                       个字符都能得到长度为n的暗黑字符串,这时候我们有结论2:1个长度为n-1的暗黑字符串可以得到3个长度为n的                         暗黑字符串;将结论1和2统一起来,我们有结论3:长度为n的暗黑字符串个数等于长度为n-1的暗黑字符串的个数                       再加上长度为n-1且最后两个字符相同的暗黑字符串个数;现在的我们要解决的问题就是长度为n-1且最后两个字符                       相同的暗黑字符串有多少个,那就要用到f(n-2)了,长度为n-2的暗黑字符串只要加上一个和最后一位字符相同                       的字符,我们便可以得到长度为n-1且最后两个字符相同的暗黑字符串,那么结论4就是,1个长度为n-2的暗黑字                         符串可以得到1个长度为n-1且最后两个字符相同的暗黑字符串;将结论3和4统一起来就是结论5,长度为n的暗黑                         字符串个数等于长度为n-1的暗黑字符串个数的2倍加上长度为n-2的暗黑字符串的个数。

猜你喜欢

转载自blog.csdn.net/weixin_41481113/article/details/83956343