![fc44319037fe3c9660baab6718c65644.png](https://i-blog.csdnimg.cn/blog_migrate/9507ef0be76d86bc3942085d4de33d51.png)
题目链接: https://leetcode-cn.com/problems/magical-string
难度:中等
通过率:49.9%
题目描述:
神奇的字符串 S 只包含 '1' 和 '2',并遵守以下规则:
字符串 S 是神奇的,因为串联字符 '1' 和 '2' 的连续出现次数会生成字符串 S 本身。
字符串 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' 的数目。
注意: N 不会超过 100,000。
示例:
输入: 6 输出: 3 解释: 神奇字符串 S 的前 6 个元素是 "12211",它包含三个 1,因此返回 3。
示例:
思路:
经过测试列表操作比字符串操作更快。
代码:
class Solution:
def magicalString(self, n: int) -> int:
s = ["1"]
i, j = 0, 0
while j < n:
if s[j] == "1":
s += ["2"] if s[-1] == "1" else ["1"]
i += 1
else:
s += ["1", "2"] if s[-1] == "1" else ["2", "1"]
i += 2
j += 1
return s[:n].count("1")