Easy
Count-and-say 序列是一個(gè)如下的整數(shù)序列。 給定一個(gè)整數(shù)n,返回第n個(gè)序列。序列用string表示。
1, 11, 21, 1211, 111221, ...
1讀作一個(gè)1,或11
11讀作兩個(gè)1,或21
21讀作一個(gè)2一個(gè)1,或1211
Leecode把此題歸為easy檔,我倒是覺得有些難度。
最開始我把題理解錯(cuò)誤,認(rèn)為是: 第一個(gè)數(shù)是n,那么由n衍生出的coutn-and-say序列的第n個(gè)值是什么?所以就有了下面的code。(開始是string=str(n))。
將問題簡(jiǎn)化為求上面序列的第n個(gè)值,應(yīng)該有更簡(jiǎn)單的辦法,歡迎反饋。
from itertools import groupby
class Solution(object):
def countAndSay(self, n):
"""
:type n: int
:rtype: str
"""
if n == 0 or n == 1:
return str(n)
string = '1'
for _ in xrange(n-1):
keys = [key for key, _ in groupby(string)]
counts = [len(list(group)) for _, group in groupby(string)]
resp = ''
for key, count in zip(keys,counts):
resp += str(count)+key
string = resp
return string