Leetcode 667.优美的排列II

优美的排列II

给定两个整数 n 和 k，你需要实现一个数组，这个数组包含从 1 到 n 的 n 个不同整数，同时满足以下条件：

① 如果这个数组是 [a₁, a₂, a₃, ... , a_n] ，那么数组 [|a₁ - a₂|, |a₂ - a₃|, |a₃ - a₄|, ... , |a_n-1 - a_n|] 中应该有且仅有 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

   

提示:

1.  n 和 k 满足条件 1 <= k < n <= 10⁴.

 

思路

按照1，k+1，2，k，3，k-1，……的顺序排列，得到的差值为k，k-1，k-2，……，1。
时间复杂度O（n）。

 

class Solution {
    public int[] constructArray(int n, int k) {
        int l=1,r=k+1;
        int[] ans=new int[n];
        int index=0;
        while(l<=r){
            ans[index++]=l++;
            if(l<=r) ans[index++]=r--;
        }
        for(int i=k+2;i<=n;i++) ans[index++]=i;
        return ans;
    }
}