您的位置:首页 > 其它

leetcode -- Gray Code -- 数学题

2015-12-15 20:43 459 查看
https://leetcode.com/problems/gray-code/

格雷码的生成方法。

思路1 数学方法

给定n,则总共有1<<<
可以让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
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: