题目解析
给定一个长度为N的整数序列,从中选出一些数,使得相邻两个选出的数之差不超过K,并使选出的数的总数最大。
解题思路
本题可以滑动窗口来解决。
首先将原数组按照从小到大排序,然后设左右指针分别指向区间的左右端点,用双指针维护区间中符合条件的数的个数。
具体流程如下:
循环遍历整个数组,将整个数组按照从小到大排序。
定义左右指针,初始值均为0。同时定义一个变量max来记录符合要求的最长子序列的长度。
利用while循环进行滑动窗口操作:
- 若arr[right] - arr[left] > k,说明当前的区间不符合要求,此时,需要将left指针向右移动,即缩小区间。
- 若arr[right] - arr[left] <= k,说明当前的区间符合要求,那么需要将right指针向右移动,即扩大区间。
在每次扩大区间时,需要更新max的值,即 max = Math.max(max, right - left + 1)。
最终输出max即可。
代码实现
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
// 注意 hasNext 和 hasNextLine 的区别
while (in.hasNextInt()) { // 注意 while 处理多个 case
int n = in.nextInt();
int k = in.nextInt();
int[] arr = new int[n];
for (int i = 0; i < n; i++) {
arr[i] = in.nextInt();
}
Arrays.sort(arr);
int max = -1,left = 0,right = 0;
while(right < n){
while(arr[right] - arr[left] > k){//1.判断条件出窗口
left++;
}
//2.更新结果
max = Math.max(max,right - left + 1);
//3.进窗口
right++;
}
System.out.println(max);
}
}
复杂度分析
时间复杂度:O(nlogn),其中n为数组的长度。因为需要进行一次排序操作,所以时间复杂度为O(nlogn)。
空间复杂度:O(n),需要使用一个数组来存储输入的n个数。