

LeetCode: 667. 优美的排列 II

• LeetCode: 667. 优美的排列 II
  • 1、题目描述
  • 2、解题思路

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 <= 104.

2、解题思路

• n个数字能够产生的间隔一共是n-1中，分别为1到n-1
• 因此，首先从大到小生成k-1个，然后剩余的按照顺序，使用最小的间隔1，最终所有间隔即为k个

class Solution:
    def constructArray(self, n: int, k: int) -> List[int]:
        left, right = 1, n
        res = []

        flag = True
        for i in range(k - 1):
            if flag:
                res.append(left)
                left += 1
            else:
                res.append(right)
                right -= 1
            flag = not flag

        if not flag:
            res.extend(reversed(range(left, right + 1)))
        else:
            res.extend(range(left, right + 1))
        return res