给你两个整数 n 和 k ,请你构造一个答案列表 answer ,该列表应当包含从 1 到 n 的 n 个不同正整数,并同时满足下述条件:
假设该列表是 answer = [a1, a2, a3, … , an] ,那么列表 [|a1 - a2|, |a2 - a3|, |a3 - a4|, … , |an-1 - an|] 中应该有且仅有 k 个不同整数。
返回列表 answer 。如果存在多种答案,只需返回其中 任意一种 。
示例 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
提示:
1 <= k < n <= 104
class Solution {
public:
vector<int> constructArray(int n, int k) {
vector<int> answer;
for (int i = 1; i < n - k; ++i) {
answer.push_back(i);
}
for (int i = n - k, j = n; i <= j; ++i, --j) {
answer.push_back(i);
if (i != j) {
answer.push_back(j);
}
}
return answer;
}
};
复杂度分析
时间复杂度:O(n)。
空间复杂度:O(1),这里不计入返回值需要的空间。