力扣 481. 神奇字符串

神奇字符串 s 仅由 '1' 和 '2' 组成，并需要遵守下面的规则：

神奇字符串 s 的神奇之处在于，串联字符串中 '1' 和 '2' 的连续出现次数可以生成该字符串。
s 的前几个元素是 s = "1221121221221121122……" 。如果将 s 中连续的若干 1 和 2 进行分组，可以得到 "1 22 11 2 1 22 1 22 11 2 11 22 ......" 。每组中 1 或者 2 的出现次数分别是 "1 2 2 1 1 2 1 2 2 1 2 2 ......" 。上面的出现次数正是 s 自身。

给你一个整数 n ，返回在神奇字符串 s 的前 n 个数字中 1 的数目。

示例 1：

输入：n = 6
输出：3
解释：神奇字符串 s 的前 6 个元素是 “122112”，它包含三个 1，因此返回 3 。 

思路：用1个列表，初始值1、2、2，然后构造这个字符串

代码

class Solution:
    def magicalString(self, n: int) -> int:
        list1 = [1,2,2]
        word = 1
        c = 0
        c1 = 0
        if n < 4:
            return 1
        for i, x in enumerate(list1):
            flag = True
            for j in range(0, x):
                c += 1
                if c > 3:
                    list1.append(word)
                if word == 1:
                    c1 += 1
                if c == n:
                    flag = False
                    break
            if flag == False:
                break
            word = 2 if word == 1 else 1

        return c1

n = 20
s = Solution()
print(s.magicalString(n))
结果