格雷编码python实现

LeetCode 链接：https://leetcode-cn.com/problems/gray-code

题目：

格雷编码是一个二进制数字系统，在该系统中，两个连续的数值仅有一个位数的差异。

给定一个代表编码总位数的非负整数 n，打印其格雷编码序列。格雷编码序列必须以 0 开头。

示例 1:

输入: 2
输出: [0,1,3,2]
解释:
00 - 0
01 - 1
11 - 3
10 - 2

思路：

设 nn 阶格雷码集合为 G(n)G(n)，则 G(n+1)G(n+1) 阶格雷码为：
给 G(n)G(n) 阶格雷码每个元素二进制形式前面添加 00，得到 G'(n)G (n)；
设 G(n)G(n) 集合倒序（镜像）为 R(n)R(n)，给 R(n)R(n) 每个元素二进制形式前面添加 11，得到 R'(n)R ′(n)；
G(n+1) = G'(n) ∪ R'(n)G(n+1)=G′(n)∪R′(n) 拼接两个集合即可得到下一阶格雷码。

 

from typing import List
class Solution:
    def grayCode(self, n: int) -> List[int]:
        g = ['' for _ in range(n+1)]
        if n==0 :   return [0]
        else:    g[0] =['0']
        if n==1 :   return [0, 1]
        else:    g[1] = ['0','1']
        h0 = lambda x: '0' + x
        h1 = lambda x: '1' + x
        two = lambda x, y: int(x) * (2 ** int(y))

        for i in range(2, n + 1):
            tmp = list(map(h0, g[i - 1]))
            value = tmp[:] + g[i-1][::-1]
            value[len(tmp):] = list(map(h1, value[len(tmp):]))
            g[i] = value

        l = [i for i in range(n)]
        r = []
        for i in range(len(g[n])):
            value = g[n][i]
            r += [sum(list(map(two, value , l[::-1])))]
        return r

 

扩展：

自己实现的二进制转整数：

s = ['1','0','0','0']
two = lambda x,y: x*(2**y)
l = [i for i in range(len(s))]
r = sum(list(map(int(two),s,l[::-1])))