leetcode -- Gray Code -- 数学题

https://leetcode.com/problems/gray-code/

格雷码的生成方法。

思路1 数学方法

给定n，则总共有1 << n个code，
可以让i从0开始，每个code就是(i/2 )^ i 即，(i>>1)^i异或。跟0异或就是本身

class Solution:
    # @return a list of integers
    def grayCode(self, n):
        res=[]
        size=1<<n
        for i in range(size):
            res.append((i>>1)^i)
        return res

思路2

我想的是backtracking的办法，如果父节点为0，则先搜0再搜1；如果父节点为1，则先搜1再搜0。但是结果顺序不对。还没找出原因

class Solution(object):

    def convertBit(self, mylist):
        mylist.reverse()
        res = 0
        for i,c in enumerate(mylist):
            res += 2 ** i * c
        return res


    def dfs(self, start, end, last_bit, subres, res):
        if start == end + 1:
            res.append(self.convertBit(subres[:]))
            return
        else:
            if last_bit == 0:
                subres[start] = 0
                self.dfs(start + 1, end, 0, subres, res)
                subres[start] = 1
                self.dfs(start + 1, end, 1, subres, res)
            else:
                subres[start] = 1
                self.dfs(start + 1, end, 1, subres, res)
                subres[start] = 0
                self.dfs(start + 1, end, 0, subres, res)

    def grayCode(self, n):
        """ :type n: int :rtype: List[int] """
        if n == 0: return [0]
        subres = [0]*n
        res = []
        self.dfs(0, n-1, 0, subres, res)
        return res