[10.31] 481. 神奇字符串

题目链接:481. 神奇字符串

题解:

解题思路

  • 技巧1:使用异或3来切换数字1和2.
    异或操作是:相同为0,不同为1
    数字1:01 ^ 11(3) = 10
    数字2:10 ^ 11(3) = 01

从题目可以知道,字符串仅能由1和2组成,故我们将数字分为包含1和2的不同组。
“1 22 11 2 1 22 1 22 11 2 11 22 …”

  • 技巧2:使用双指针的策略,一个指针tail指向下一组数据的开头,另一个指针point指向当前组数据需要的个数。

算法流程:

  1. 初始化:首先我们知道第一个数和第二个数一定是1和2,所以我们定义一个长为n的list,并初始化第一个元素为1.
magic_list = [0] * n
# 初始化第一个元素
magic_list[0] = 1

我们知道第二个数字一定是2,并且这个2一定会有两个,所以将tail和point,分别做如下的初始化.

在这里插入图片描述

需要注意的是,如果n<2,则结果一定为1,如果不直接输出则会出现list out of range的错误。
if n <= 2: return 1

  1. 填第2组数据,从初始化中我们知道第二组中含有两个2.因此我们首先将填入的数据由1转换成2,value ^= 3.其次在对应的位置填入数据.

在这里插入图片描述
更新此时下一组数字应该出现的次数,count = magic_list[point]
接下来的更新如下图所示:
在这里插入图片描述
在这里插入图片描述

代码

class Solution:
    def magicalString(self, n: int) -> int:
        magic_list = [0] * n
        # 初始化第一个元素
        magic_list[0] = 1
        count = 2  # 第二组数字是2,因此count 记为2
        tail, result, value, point = 1, 1, 1, 2
        if n <= 2: return 1

        while tail < n:
            value ^= 3  # 异或
            while count > 0 and tail < n:
                count -= 1
                magic_list[tail] = value
                tail += 1
                if value == 1:
                    result += 1
            count = magic_list[point]
            point += 1
        
        return result

猜你喜欢

转载自blog.csdn.net/leeyns/article/details/127615995