LeetCode打卡：667. 优美的排列 II

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

链接：https://leetcode.com/problems/beautiful-arrangement-ii/
解题思路：（振荡数列）
对前k+1个数，让其满足相邻元素之差的绝对值正好是k个数，若n>k+1,则将k+2~n依次添加到后面。

class Solution(object):
    def constructArray(self, n, k):
        """
        :type n: int
        :type k: int
        :rtype: List[int]
        """
        # 阻尼振荡数组
        if k == 1:
            return range(1, n + 1)
        res = [1]
        cur = 1  # 振荡方向
        j = k    # 振荡幅度
        while j > 0:
            res.append(res[-1] + cur * j)
            j -= 1
            cur *= -1
            
        # k个不同整数已满足，将剩余k + 2~n 的数字按等差数列添加到后面   
        j = k + 1
        while j < n:
            res.append(j + 1)
            j += 1
        return res

Time: O(N)
Space: O(1)
难度：中等
标签：数组