leetcode:count-and-say [外观数列] 递归法

本题主要考察 动态规划字符串的处理

'''
「外观数列」是一个整数序列,从数字 1 开始,序列中的每一项都是对前一项的描述。前五项如下:
1.     1
2.     11
3.     21
4.     1211
5.     111221
1 被读作  "one 1"  ("一个一") , 即 11。
11 被读作 "two 1s" ("两个一"), 即 21。
21 被读作 "one 2",  "one 1" ("一个二" ,  "一个一") , 即 1211。

给定一个正整数 n(1 ≤ n ≤ 30),输出外观数列的第 n 项。
注意:整数序列中的每一项将表示为一个字符串。
'''
"""
示例 1:
输入: 1
输出: "1"
解释:这是一个基本样例。

示例 2:
输入: 4
输出: "1211"
解释:当 n = 3 时,序列是 "21",其中我们有 "2" 和 "1" 两组,"2" 可以读作 "12",也就是出现频次 = 1 而 值 = 2;类似 "1" 可以读作 "11"。所以答案是 "12" 和 "11" 组合在一起,也就是 "1211"。
"""

这道题,我第一想到的是使用递归,在解题过程中感觉对递归还是用的不够熟练,边界条件需要注意,最后在多次debug调试下才发现代码存在的问题在哪里,然后解决。
Gitee

class Solution:
    def countAndSay(self, n: int) -> str:
        if n == 1:
            return '1'
        pre = self.countAndSay(n - 1)
        res = ''
        counter = 1
        for i in range(len(pre)):
            # 预先处理最后一个字符,避免下标越界:只需处理最后两个不同的情况
            if i == len(pre) - 1:
                res += f'1{pre[-1]}'
            else:
                if (i + 1 == len(pre) - 1) and pre[i] == pre[i + 1]:
                    # 判断倒数第二个是否等于最后一个,如果相等,返回;若不等,留给前面的预先处理来保证
                    counter += 1
                    return res + f'{counter}{pre[i]}'
                elif pre[i] == pre[i + 1]:
                    counter += 1
                else:
                    # 下一个字符与当前不同,所以要统计当前相同的字符数,并将counter清零
                    res += f'{counter}{pre[i]}'
                    counter = 1
        return res

if __name__ == '__main__':
    s = Solution()
    print([s.countAndSay(n) for n in range(1, 7)])
    print([s.countAndSay2(n) for n in range(1, 7)])
    # 11 21 1211 111221  312211

看到另一种方法,不使用递归:只记录前后两组不同的字符,当时我看了想了好一会儿才明白。

class Solution:
    def countAndSay2(self, n: int) -> str:
        pre = '1'
        for i in range(1, n):
            res = ''
            cmp = pre[0]
            count = 1
            for j in range(1, len(pre)):
                if pre[j] == cmp:
                    count += 1
                else:
                    res += str(count) + cmp
                    cmp = pre[j]
                    count = 1
            res += str(count) + cmp
            pre = res
        return pre

发布了91 篇原创文章 · 获赞 47 · 访问量 5万+

猜你喜欢

转载自blog.csdn.net/qq_31362767/article/details/105350224