列表

详情


481. 神奇字符串

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

s 的前几个元素是 s = "1221121221221121122……" 。如果将 s 中连续的若干 12 进行分组,可以得到 "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

 

提示:

原站题解

去查看

上次编辑到这里,代码来自缓存 点击恢复默认模板
class Solution { public: int magicalString(int n) { } };

python3 解法, 执行用时: 232 ms, 内存消耗: 21.9 MB, 提交时间: 2022-10-31 09:24:10

class Solution:
    def magicalString(self, n: int) -> int:
        if n < 4:
            return 1
        s = [''] * n
        s[:3] = "122"
        res = 1
        i, j = 2, 3
        while j < n:
            size = int(s[i])
            num = 3 - int(s[j - 1])
            while size and j < n:
                s[j] = str(num)
                if num == 1:
                    res += 1
                j += 1
                size -= 1
            i += 1
        return res

python3 解法, 执行用时: 112 ms, 内存消耗: 16.4 MB, 提交时间: 2022-10-31 09:23:00

class Solution:
    def magicalString(self, n: int) -> int:
        s = [1, 2, 2] # 分组
        i = 2
        while len(s) < n:
            s += [s[-1] ^ 3] * s[i]  # 1^3=2, 2^3=1,这样就能在 1 和 2 之间转换
            i += 1
        return s[:n].count(1)

上一题