【LeetCode】481. 神奇字符串 解题报告 (python)

原题地址：https://leetcode-cn.com/problems/magical-string/

题目描述：

神奇的字符串 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:
        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")

class Solution:
    def magicalString(self, n: int) -> int:
        s='122'
        towrite=2
        while len(s)<n:
            if s[towrite]=='1':
                s+='2' if s[-1]=='1' else '1'
            else:
                s+='22' if s[-1]=='1' else '11'
            towrite+=1
        return s[:n].count('1')