LeetCode 1238.循环码排列

最新推荐文章于 2024-08-15 05:02:54 发布

Syliva_

最新推荐文章于 2024-08-15 05:02:54 发布

阅读量74

点赞数

分类专栏： LeetCode每日一题文章标签： c++ 数据结构

本文链接：https://blog.csdn.net/weixin_50190226/article/details/129188048

版权

LeetCode每日一题专栏收录该内容

31 篇文章 0 订阅

订阅专栏

文章介绍了如何使用格雷码的特性来解决LeetCode上的一个排列问题，即生成满足特定条件的排列。通过从0开始生成格雷码，找到给定的起始值start，然后进行位操作，实现了从start开始的排列。文中提到的算法时间复杂度为O(2^n)，空间复杂度同样为O(2^n)，并提出使用数学公式优化，降低了时间和空间需求。

摘要由CSDN通过智能技术生成

题目描述

给你两个整数 n 和 start。你的任务是返回任意 (0,1,2,…,2^n-1) 的排列 p，并且满足：

p[0] = start
p[i] 和 p[i+1] 的二进制表示形式只有一位不同
p[0] 和 p[2^n -1] 的二进制表示形式也只有一位不同

示例 1：

输入：n = 2, start = 3
输出：[3,2,0,1]
解释：这个排列的二进制表示是 (11,10,00,01)
     所有的相邻元素都有一位是不同的，另一个有效的排列是 [3,1,0,2]

示例 2：

输出：n = 3, start = 2
输出：[2,6,7,5,4,0,1,3]
解释：这个排列的二进制表示是 (010,110,111,101,100,000,001,011)

提示：

1 <= n <= 16
0 <= start < 2^n

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/circular-permutation-in-binary-representation

题目分析

觉得这个定义非常眼熟，仔细一想觉得很像计算机组成原理里学到的格雷编码。那就顺便复习一下：
格雷码的特点是除去最高位后，剩下几位镜像对称，可以利用该特点递推快速生成n位格雷码。
采用策略：先从0开始生成格雷码，再寻找start的位置，移位后返回结果

class Solution {
public:
    vector<int> circularPermutation(int n, int start) {
        int length = pow(2,n);
        string grey[length];
        vector<int> temp(length);

        for(int i=1;i<=n;i++){
            if(i==1){
                grey[0] = '0';
                grey[1] = '1';
            }
            else{
                for(int j=0;j<pow(2,i-1);j++){
                    int t=pow(2,i-1)-1; //当前最后一个数
                    grey[t+j+1]='1'+grey[t-j];
                    grey[t-j]='0'+grey[t-j];  
                }
            }
        }
        int flag=0; //标记开始位置
        for(int i=0;i<length;i++)
        {
            //cout<<grey[i]<<endl;
            for(int j=0;j<n;j++){
                if(grey[i][j]=='1'){
                    temp[i]+=pow(2,n-1-j);
                }
            }
            if(temp[i]==start) flag = i; //把ans数组向左移i位
        }

        vector<int> ans(length);
        for(int i=0;i<length;i++){
            ans[i]=temp[(flag+i)%length]; //复制数组
        }
        return ans;
    }
};

虽然AC但是时间空间都存在可优化之处，时间复杂度为 $O(2^n)$ ，空间复杂度也是 $O(2^n)$
在这里插入图片描述
官方题解
进一步利用数学性质，用公式法求格雷码，与start做按位异或运算实现从start开始。在这里记一下公式：


class Solution {
public:
    vector<int> circularPermutation(int n, int start) {
        int nn = 1<<n;
        vector<int> res(nn, 0);
        for (int i = 0; i < nn; ++i)
        {
            res[i] = i ^ (i>>1) ^ start;
        }

        return res;
    }
};