【leetcode刷题】20T45-格雷编码

示例 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 的格雷编码序列，其长度为 2^n。当 n = 0 时，长度为 2^0 = 1。
     因此，当 n = 0 时，其格雷编码序列为 [0]。

class Solution(object):
    def grayCode(self, n):
        """
        :type n: int
        :rtype: List[int]
        """
        if n == 0:
            return [0]
        ls = ['0', '1']
        for i in range(1, n):
            tmp1 = list(map(lambda x: '0' + x, copy.copy(ls)))
            tmp2 = list(map(lambda x: '1' + x, copy.copy(ls[::-1])))
            ls = tmp1 + tmp2

        ls = list(map(lambda x: int(x, 2), ls))
        return ls

class Solution {
public:
    vector<int> grayCode(int n) {
        vector<int> res;
        // 特殊情况
        res.push_back(0);
        if (n == 0)
            return res;
        res.push_back(1);
        if (n == 1)
            return res;

        int add = 1;
        for (int i = 1; i < n; i++) {
            add *= 2;
            for (int j = res.size() - 1; j >= 0; j--) {
                res.push_back(res[j] + add);
            }
        }
        return res;
    }
};