给定一个代表编码总位数的非负整数 n,打印其格雷编码序列。格雷编码序列必须以 0 开头。
示例 1:
输入: 2
输出: [0,1,3,2]
解释:
00 - 0
01 - 1
11 - 3
10 - 2
对于给定的 n,其格雷编码序列并不唯一。
例如,[0,2,3,1] 也是一个有效的格雷编码序列。
00 - 0
10 - 2
11 - 3
01 - 1
示例 2:
输入: 0
输出: [0]
解释: 我们定义格雷编码序列必须以 0 开头。
给定编码总位数为 n 的格雷编码序列,其长度为 2n。当 n = 0 时,长度为 20 = 1。
因此,当 n = 0 时,其格雷编码序列为 [0]。
这个题只要知道格雷码与二进制数之间的转换即可,二进制数转换成格雷码只需要本身与其右移一位相异或即可。
这里可以直接右移异或,因为如果最高位为0,右移之后最高位也是0,异或还是0,如果最高位为1,右移之后最高位为0,异或也还是1,所以不需要特殊处理最高位。
C++源代码:
class Solution {
public:
vector<int> grayCode(int n) {
vector<int> res;
for(int i=0;i<pow(2, n);i++)
res.push_back((i >> 1) ^ i);
return res;
}
};
python3源代码:
class Solution:
def grayCode(self, n):
"""
:type n: int
:rtype: List[int]
"""
res = []
for i in range(pow(2, n)):
res.append((i >> 1) ^ i)
return res