1. 题目
给定两个整数 n 和 k,你需要实现一个数组,这个数组包含从 1 到 n 的 n 个不同整数,同时满足以下条件:
① 如果这个数组是 [a1, a2, a3, ... , an]
,那么数组 [|a1 - a2|, |a2 - a3|, |a3 - a4|, ... , |an-1 - an|]
中应该有且仅有 k 个不同整数;.
② 如果存在多种答案,你只需实现并返回其中任意一种.
示例 1:
输入: n = 3, k = 1
输出: [1, 2, 3]
解释: [1, 2, 3] 包含 3 个范围在 1-3 的不同整数, 并且 [1, 1] 中有且仅有 1 个不同整数 : 1示例 2:
输入: n = 3, k = 2
输出: [1, 3, 2]
解释: [1, 3, 2] 包含 3 个范围在 1-3 的不同整数, 并且 [2, 1] 中有且仅有 2 个不同整数: 1 和 2提示:n 和 k 满足条件 1 <= k < n <= 10^4.
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/beautiful-arrangement-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 找规律,翻转法
class Solution {
public:vector<int> constructArray(int n, int k) {int i;vector<int> ans(n);for(i = 0; i < n; ++i)ans[i] = i+1;for(i = 1; i < k; ++i)reverse(ans,i,n-1);return ans;}void reverse(vector<int>& arr, int i, int j){while(i < j)swap(arr[i++], arr[j--]);}
};
- 优化,直接填写数字
class Solution {
public:vector<int> constructArray(int n, int k) {int i, d = n-1;bool flag = true;vector<int> ans(n,1);for(i = 1; i < k; ++i,--d){if(flag)ans[i] = ans[i-1]+d;elseans[i] = ans[i-1]-d;flag = !flag;}d = 1;for( ; i < n; ++i){if(flag)ans[i] = ans[i-1]+d;elseans[i] = ans[i-1]-d;}return ans;}
};