题目描述:
对于某些固定的 N,如果数组 A 是整数 1, 2, ..., N
组成的排列,使得:
对于每个 i < j
,都不存在 k 满足 i < k < j
使得 A[k] * 2 = A[i] + A[j]
。
那么数组 A 是漂亮数组。
给定 N,返回任意漂亮数组 A(保证存在一个)。
示例 1:
输入:4
输出:[2,1,4,3]
示例 2:
输入:5
输出:[3,1,2,5,4]
提示:
1 <= N <= 1000
解法:
class Solution {
public:
vector<int> beautifulArray(int N) {
if(N == 1){
return {1};
}else if(N == 2){
return {1, 2};
}else if(N == 3){
return {1, 3, 2};
}else{
vector<int> res = {1, 3, 2};
int sz = 3;
while(sz < N){
for(auto& val : res){
val *= 2;
}
for(int i = 0; i < sz; i++){
res.push_back(res[i]-1);
}
sz *= 2;
}
if(sz > N){
int i = 0;
while(i < sz){
if(res[i] > N){
res.erase(res.begin() + i);
}else{
i++;
}
}
}
return res;
}
}
};