leetcode 932. 漂亮数组(Beautiful Array)

目录

• 题目描述：
• 示例 1：
• 示例 2：
• 解法：

题目描述：

对于某些固定的 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;
        }
    }
};