0数学中等 LeetCode667. 优美的排列 II

667. 优美的排列 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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/beautiful-arrangement-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

1，n, 2, n-1, 3, n-2, 4, n-3, 5,…
按照上面的规律持续下去能够找到最多的不同正整数。

class Solution {
    public int[] constructArray(int n, int k) {
        int[] ans = new int[n];
        int index = 0;
        for (int i = 0; i < n / 2 && k > 0; i++) {
            ans[index++] = i + 1;
            --k;
            if (k > 0) {
                ans[index++] = n - i;
                k--;
            }
        }
        if (index < n) {
            if (index % 2 == 0) {
                while (index != n) {
                    ans[index] = ans[index-1] - 1;
                    index++;
                }
            } else {
                while (index != n) {
                    ans[index] = ans[index-1] + 1;
                    index++;
                }
            }
        }
        return ans;
    }
}