题目链接: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和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
- 填第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