题目描述
给你两个整数 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)
O(2n),空间复杂度也是
O
(
2
n
)
O(2^n)
O(2n)
官方题解
进一步利用数学性质,用公式法求格雷码,与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;
}
};
总结
吁,至少看出来了是格雷码,虽然公式不记得了(惭愧)。把上述代码中的pow全部替换成移位运算后,时间上也得到了一定的提升。在能力范围内尽量多使用数学方法,有利于降低时间和空间复杂度。