【美团春招20230304】1.平滑值 ＜模拟＞

【美团春招20230304】1.平滑值

小红定义一个数组的平滑值为:任意两个相邻元素的差的绝对值的最大值。
例如 : 数组 [2,4,3,3] 的平滑值为 |2-4| =2。
小红希望你构造一个长度为 n 的排列，满足排列的平滑值等于 k。你能帮帮她吗? 排列是指:长度为 n 的数组，1 到 n 每个正整数都恰好出现 1 次。

输入描述
第一行输入两个正整数，分别表示n和k。其中，1≤ k < n ≤$10^5$

输出描述
输出n个数字，用空格隔开。

输入

 4 2

输出

1 3 2 4  

题解

直接找规律，先直接生成正序全排列，然后把 k 提到序列头

import java.util.ArrayList;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int k = sc.nextInt();

        ArrayList<Integer> arr = new ArrayList<>();

        arr.add(k + 1);
        for (int i = 1; i <= k; i++) {
            arr.add(i);
        }
        for (int j = k + 2; j <= n; j++) {
            arr.add(j);
        }

        for (int i : arr) {
            System.out.print(i + " ");
        }
    }
}

其中存在 k 为 1 时出错，改良代码如下：

import java.util.ArrayList;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int k = sc.nextInt();

        ArrayList<Integer> arr = new ArrayList<>();

        if(k<n) {
            if(k==1){
                for (int i = 1; i <= n; i++) {
                    System.out.print(i + " ");
                }
                return;
            }
            arr.add(k + 1);
            for (int i = 1; i <= k; i++) {
                arr.add(i);
            }
            for (int j = k + 2; j <= n; j++) {
                arr.add(j);
            }

            for (int i : arr) {
                System.out.print(i + " ");
            }
        }
        else {
            System.out.println("-1");
        }
    }
}