题目

思路

首先我们先看一下格雷编码的一些情况,为了一会方便理解,我们看它的二进制情况。

当n=1时,输出[0,1]

当n=2时,输出[00,01,11,10]

当n=3时,输出[000, 001, 011, 010, 110, 111, 101, 100]

我们可以看到2的前半部分就是1,3的前半部分就是2。

所以我们只需要求后半部分即可,

格雷码的生成有一个递推公式。已知n-1位的格雷码,如何得到n位的?方法是,首先将n-1位的格雷码列表逆序,然后每个数的最高位设为1,然后拼接到原来的列表后面。例如,n=1的格雷码是0,1。n=2的时候,将之前的逆序是1,0,然后前面加上最高位1,变成11,10,然后拼接到原来的00,01后面,得到00,01,11,10,对应十进制的0,1,3,2。

总结一下,就是:

  • 将k-1位序列逆序。

  • 对逆序后的每个元素,在最高位(即第k-1位)添加1。

  • 将新生成的序列追加到原序列之后。

代码详解

首先,定义一个ans用来存储格雷码序列。

ans.reserve是将ans的capacity容量扩大,n位格雷码有2^n个序列,所以就给他2^n的空间。

至于里面为什么是1<<n。

它的意思就是把1向左移动n位

n=1, 移动完:10              2^1=2

n=2, 移动完:100            2^2=4

n=3, 移动完:1000          2^3=8

接着将里面的变量初始化为0。

for(int i=1;i<=n;i++) 这个是用来逐位构造格雷码的,对于n=1,只会循环一次

int m =ans.size();

这个是求ans现有的元素数

for(int j=m-1;j>=0;j--)
            {
                ans.push_back(ans[j] | (1 << (i-1)));
            }

这个就是对里面现有的数进行倒序访问。

ans.push_back(ans[j] | (1 << (i-1)));

这个是1左移i-1位与原有的数相或。如果i=1,就是1左移0位,其实他就是用来将第i-1位设为1,并添加到结果中。

当n=1时,输出[0,1]

当n=2时,输出[00,01,11,10]

对于这个例子,n=2就是将第二位设为1,分别是11,10并添加到序列中,因为是逆序,所以先给1加,后给0加。

最后返回ans即可。

代码

class Solution {
public:
    vector<int> grayCode(int n) {
        vector<int> ans;
        ans.reserve(1<<n);
        ans.push_back(0);
        for(int i=1;i<=n;i++)
        {
            int m =ans.size();
            for(int j=m-1;j>=0;j--)
            {
                ans.push_back(ans[j] | (1 << (i-1)));
            }
        }
        return ans;
    }
};

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐