LeetCode解題之Gray Code

原題

格雷碼表示在一組數的編碼中，若任意兩個相鄰的代碼只有一位二進制數不同。現給定二進制碼的位數，要求打印出格雷碼序列。

注意點：

例子：

輸入: n = 2

輸出: [0,1,3,2]

00 - 0
01 - 1
11 - 3
10 - 2

解題思路

根據維基百科上的關於 格雷碼和二進制數的轉換關系 實現的代碼。

AC源碼

class Solution(object):
class Solution(object):
    def grayCode(self, n):
        """
        :type n: int
        :rtype: List[int]
        """
        result = [(i >> 1) ^ i for i in range(pow(2, n))]
        return result


if __name__ == "__main__":
    assert Solution().grayCode(2) == [0, 1, 3, 2]