n 位格雷码序列 是一个由 2n 个整数组成的序列,其中:
每个整数都在范围 [0, 2n - 1] 内(含 0 和 2n - 1)
第一个整数是 0
一个整数在序列中出现 不超过一次
每对 相邻 整数的二进制表示 恰好一位不同 ,且
第一个 和 最后一个 整数的二进制表示 恰好一位不同
给你一个整数 n ,返回任一有效的 n 位格雷码序列 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/gray-code
例:
输入:n = 2
输出:[0,1,3,2]
解释:
[0,1,3,2] 的二进制表示是 [00,01,11,10] 。
- 00 和 01 有一位不同
- 01 和 11 有一位不同
- 11 和 10 有一位不同
- 10 和 00 有一位不同
[0,2,3,1] 也是一个有效的格雷码序列,其二进制表示是 [00,10,11,01] 。
- 00 和 10 有一位不同
- 10 和 11 有一位不同
- 11 和 01 有一位不同
- 01 和 00 有一位不同
解析:
将格雷编码对称翻转然后加上2^(n-1),会发现两组数据首尾只有一位不同,利用这个规律进行编写即可。
class Solution(object):
def grayCode(self, n):
"""
:type n: int
:rtype: List[int]
"""
res = [0, 1]
if n == 1:
return res
for i in range(1, n):
sum = list(reversed(res)) # 翻转
for item in sum:
item += 2**i # 加上2^(n-1)
res.append(item) # 将增加后的值加入到格雷编码中
return res