leetcode_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 。 

示例 2：

输入：n = 1
输出：1

提示：

• 1 <= n <= 105

核心代码

class Solution:
    def magicalString(self, n: int) -> int:
        if n == 0:
            return 0
        if n < 3:
            return 1

        s = "122"
        a = "1"
        i = 2
        tmp_n = 3
        while tmp_n <= n:
            s += a * int(s[i])
            tmp_n += int(s[i])
            a = "2" if a == "1" else "1"
            i += 1
        return s[:n].count("1")

解题思路：我们首先就是要明白神奇字符串的构造的方式，核心1，2作为构造字符循环出现，然后我们已经构造好的字符中位置决定了1，2的构造字符出现的个数，直接统计1出现的个数比较简单，重点在神奇字符串的构建。