1.题目
LeetCode: 667. 优美的排列 II
【medium】
给定两个整数 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.$
2.解题
方法一:
让前k+1个元素构建出k个不同的差值,然后将剩余的数组补齐。
java:
class Solution {
public int[] constructArray(int n, int k) {
int[] res = new int[n];
res[0] = 1;
for (int i = 1, j = k; i <= k; i++, j--) {
res[i] = i % 2 == 1 ? res[i - 1] + j : res[i - 1] - j;
}
for (int i = k + 1; i < n; i++) {
res[i] = i + 1;
}
return res;
}
}
时间复杂度:O(n)
空间复杂度:O(n)