【美团春招20230304】1.平滑值
小红定义一个数组的平滑值为:任意两个相邻元素的差的绝对值的最大值。
例如 : 数组 [2,4,3,3] 的平滑值为 |2-4| =2。
小红希望你构造一个长度为 n 的排列,满足排列的平滑值等于 k。你能帮帮她吗? 排列是指:长度为 n 的数组,1 到 n 每个正整数都恰好出现 1 次。
输入描述
第一行输入两个正整数,分别表示n和k。其中,1≤ k < n ≤
1
0
5
10^5
105
输出描述
输出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");
}
}
}