神奇字符串中 1 的个数 - Python 代码实现

神奇字符串中 1 的个数 - Python 代码实现

神奇字符串 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 的数目。

以下是根据题目要求计算前 n 个数字中 1 的数目的 Python 代码：

def magical_string(n):
    s = '122'
    p = 2  # 指向当前需要扩展的数
    while len(s) < n:
        if s[p] == '1':
            s += '2' * int(s[len(s) - 1])
        else:
            s += '1' * int(s[len(s) - 1])
        p += 1

    return s[:n].count('1')

n = int(input())
count_ones = magical_string(n)
print(count_ones)

代码解析：

1. 初始化字符串 s 为 '122'，p 指向当前需要扩展的数的位置，初始为 2。
2. 循环直到字符串 s 的长度大于等于 n。
3. 如果 s[p] 为 '1'，则在 s 末尾添加 '2' 乘以 s[len(s)-1] 的次数，否则添加 '1' 乘以 s[len(s)-1] 的次数。
4. 将 p 加 1，指向下一个需要扩展的位置。
5. 最后，返回字符串 s 前 n 个字符中 '1' 的个数。

希望这份代码能够帮助你理解神奇字符串的生成规则并解决这个问题。