数组题目：优美的排列 II

题目

标题和出处

标题：优美的排列 II

出处：667. 优美的排列 II

难度

7 级

题目描述

要求

给你两个整数 $\text{n}$ 和 $\text{k}$，请你构造一个答案列表 $\text{answer}$，该列表应当包含从 $\text{1}$ 到 $\text{n}$ 的 $\text{n}$ 个不同正整数，并同时满足下述条件：

• 假设该列表是 $\text{answer}=[{\text{a}}_{\text{1}},{\text{a}}_{\text{2}},{\text{a}}_{\text{3}},\dots ,{\text{a}}_{\text{n}}]$，那么列表 $[|{\text{a}}_{\text{1}}-{\text{a}}_{\text{2}}|,|{\text{a}}_{\text{2}}-{\text{a}}_{\text{3}}|,|{\text{a}}_{\text{3}}-{\text{a}}_{\text{4}}|,\dots ,|{\text{a}}_{\text{n-1}}-{\text{a}}_{\text{n}}|]$ 中应该有且仅有 $\text{k}$ 个不同整数。

返回列表 $\text{answer}$。如果存在多种答案，只需返回其中任意一种。

示例

示例 1：

输入：$\text{n = 3, k = 1}$
输出：$\text{[1, 2, 3]}$
解释：$\text{[1, 2, 3]}$ 包含 $\text{3}$ 个范围在 $\text{1}$ 到 $\text{3}$ 的不同整数，并且 $\text{[1, 1]}$ 中有且仅有 $\text{1}$ 个不同整数：$\text{1}$

示例 2：

输入：$\text{n = 3, k = 2}$
输出：$\text{[1, 3, 2]}$
解释：$\text{[1, 3, 2]}$ 包含 $\text{3}$ 个范围在 $\text{1}$ 到 $\text{3}$ 的不同整数，并且 $\text{[2, 1]}$ 中有且仅有 $\text{2}$ 个不同整数：$\text{1}$ 和 $\text{2}$

数据范围

• $\text{1}\le \text{k}<\text{n}\le {\text{10}}^{\text{4}}$

解法

思路和算法

为了方便叙述，将列表 $[|a_1-a_2|,|a_2-a_3|,|a_3-a_4|,\dots ,|a_{n-1}-a_n|]$ 称为「相邻绝对差列表」。

对于从 $1$ 到 $n$ 的任意排列，对应的「相邻绝对差列表」中的不同整数最少为 $1$ 个，即所有的相邻元素绝对差都相等，最多为 $n-1$ 个，即所有的相邻元素绝对差都不相等。

对于列表 $[1,2,3,\dots ,n]$，「相邻绝对差列表」中的元素都是 $1$，因此有 $1$ 个不同整数。

对于列表 $[1,n,2,n-1,3,n-2,\dots ]$，「相邻绝对差列表」中的元素各不相同，分别为 $n-1$ 到 $1$，因此有 $n-1$ 个不同整数。

为了让「相邻绝对差列表」中的不同整数为 $k$ 个，可以从整数 $1$ 到 $k+1$ 构造出「相邻绝对差列表」中的元素各不相同的序列：$[1,k+1,2,k,3,k-1,\dots ]$，然后将从 $k+2$ 到 $n$ 的整数依次添加到该序列的末尾。

按照上述方法构造出的序列，前 $k+1$ 个数的相邻元素绝对差各不相同，分别为 $k$ 到 $1$，从 $k+2$ 到 $n$ 的相邻元素绝对差都是 $1$，$k+2$ 前面的相邻元素一定大于 $1$，因此 $k+2$ 和前面的相邻元素的绝对差一定不超过 $k$。即整个序列的「相邻绝对差列表」包含从 $1$ 到 $k$ 的全部整数，且不包含大于 $k$ 的整数，因此「相邻绝对差列表」中有且仅有 $k$ 个不同整数。

证明

对于列表 $[1,n,2,n-1,3,n-2,\dots ]$，「相邻绝对差列表」中的元素各不相同，分别为 $n-1$ 到 $1$，有 $n-1$ 个不同整数。该结论证明如下。

用 $\text{arr}$ 表示列表 $[1,n,2,n-1,3,n-2,\dots ]$，则 $\text{arr}$ 中的每个下标对应的元素如下：

$$\text{arr}[i]=\{\begin{array}{ll}\frac{i}{2}+1,& i是偶数\\ n-\frac{i-1}{2},& i是奇数\end{array}$$

当 $1\le i<n$ 时，$\text{arr}[i]-\text{arr}[i-1]$ 计算如下：

• 当 $i$ 是奇数时：

$$\begin{array}{rl}& \text{arr}[i]-\text{arr}[i-1]\\ & =(n-\frac{i-1}{2})-(\frac{i-1}{2}+1)\\ & =n-(i-1)-1\\ & =n-i\end{array}$$

• 当 $i$ 是偶数时：

$$\begin{array}{rl}& \text{arr}[i-1]-\text{arr}[i]\\ & =(n-\frac{i-2}{2})-(\frac{i}{2}+1)\\ & =n-(\frac{i}{2}-1)-\frac{i}{2}-1\\ & =n-\frac{i}{2}+1-\frac{i}{2}-1\\ & =n-i\end{array}$$

由于 $1\le i<n$，因此 $1\le n-i\le n-1$，$\text{arr}$ 对应的「相邻绝对差列表」中的元素各不相同，分别为 $n-1$ 到 $1$。

代码

class Solution {
    public int[] constructArray(int n, int k) {
        int[] answer = new int[n];
        int low = 1, high = k + 1;
        for (int i = 0; i <= k; i++) {
            if (i % 2 == 0) {
                answer[i] = low++;
            } else {
                answer[i] = high--;
            }
        }
        for (int i = k + 1; i < n; i++) {
            answer[i] = i + 1;
        }
        return answer;
    }
}

复杂度分析

• 时间复杂度：$O(n)$。构造数组 $\text{answer}$ 的过程为遍历该数组一次，填入每个元素的时间为 $O(1)$，因此时间复杂度为 $O(n)$。

• 空间复杂度：$O(1)$。除了返回值以外，使用的空间复杂度是常数。