中等

问题描述

给你两个整数 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

构造

构造差值从 $k$ 到 $1$ 的序列，由于 $k < n$，所以必然能够构造成功，且构造所用到的数为$[1, k + 1]$，构造后所剩下的值[k + 2, n] 只需从小到大即可。

/**
 * @param {number} n
 * @param {number} k
 * @return {number[]}
 */
var constructArray = function (n, k) {
  const ans = []
  let i = 1
  while (ans.length !== n) {
    if (i <= k) {
      if (k - (i - 1) + 1 > i) {
        ans.push(i)
        ans.push(k - (i - 1) + 1)
      } else {
        if (i <= k && ans[ans.length - 1] !== i) ans.push(i)
        i = k + 1
        continue
      }
    }
    if (i > k + 1) {
      ans.push(i)
    }
    i++
  }
  return ans
}

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$