1. 问题描述:
神奇的字符串 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。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/magical-string
2. 思路分析:
首先需要读懂题目,由题目可知我们将原字符串的出现次数连接起来就可以得到原来的字符串,并且可以发现一个规律是原字符串中"1"和"2"是间隔出现的,所以我们模拟整个过程即可,根据当前的字符的次数生成长度大于等于n的字符串,最后遍历一下生成的长度为n的字符串中统计字符"1"出现的次数即可。
3. 代码如下:
class Solution:
# 先生成长度为n的字符串然后统计一下1出现的次数即可, 可以使用双指针来解决
def magicalString(self, n: int) -> int:
s = "122112"
t = "1221"
j = len(t)
while len(s) < n:
c = s[j]
if s[-1] == "2":
# 原字符串中"1"和"2"是交替出现的原字符串中重复这么多次1
s += int(c) * "1"
else:
s += int(c) * "2"
# 指向出现次数的指针往后一个位置
j += 1
res = 0
# print(s)
# 在生成的字符串中统计出现的字符"1"的数目
for i in range(n):
if s[i] == "1": res += 1
return res
class Solution:
def magicalString(self, n: int) -> int:
s = "122"
i = 2
k = 1
while len(s) < n:
s += int(s[i]) * str(k)
# 将1变成2将2变成1
k = 3 - k
i += 1
res = 0
# print(s)
for i in range(n):
if s[i] == "1": res += 1
return res