667. 优美的排列 II（难度：中等）

题目链接：https://leetcode.cn/problems/beautiful-arrangement-ii/

题目描述

给你两个整数 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时，数组[1,2,3,…,n-1,n]，相邻的两个数都是差1。
• 当k=n-1时，数组[1,n,2,n-1,3,n-3,…]，相邻的两个数的差从n-1一直递减为1。

一般情况，我们可以让前半段的差都为1，后半段的差按照第二种特殊情况，依次递减。

即：[1,2,⋯,n−k,n,n−k+1,n−1,n−k+2,⋯]

代码：

class Solution {
    public int[] constructArray(int n, int k) {
        int[] answer = new int[n];
        int idx = 0;
        for (int i = 1; i < n - k; ++i) {
            answer[idx] = i;
            ++idx;
        }
        for (int i = n - k, j = n; i <= j; ++i, --j) {
            answer[idx] = i;
            ++idx;
            if (i != j) {
                answer[idx] = j;
                ++idx;
            }
        }
        return answer;
    }
}