牛客網線上程式設計(19):暗黑字串-python
阿新 • • 發佈:2018-11-11
題目描述
一個只包含'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.更新規則: 表示輸入的整數為n時的暗黑字串數;
在計算