二面 题没做出来
问了ai感觉有道理 记录一下
要在一个数组中找到所有两两差值绝对值中的第M小的数,可以通过二分查找的方法来实现。这个方法的思路是:通过二分查找确定一个差值阈值,然后利用双指针或者其他方法计算数组中与这个阈值相等或更小的差值的个数,从而判断这个阈值是否是第M小的差值。
以下是Java实现的代码:
java
复制
import java.util.Arrays;
public class KthSmallestPairDistance {
public static int smallestDistancePair(int[] nums, int k) {
Arrays.sort(nums);
int left = 0;
int right = nums[nums.length - 1] - nums[0];
while (left < right) {
int mid = left + (right - left) / 2;
if (countPairs(nums, mid) < k) {
left = mid + 1;
} else {
right = mid;
}
}
return left;
}
private static int countPairs(int[] nums, int mid) {
int count = 0;
int left = 0;
for (int right = 0; right < nums.length; right++) {
while (nums[right] - nums[left] > mid) {
left++;
}
count += right - left;
}
return count;
}
public static void main(String[] args) {
int[] nums = {1, 3, 1};
int k = 1;
System.out.println(smallestDistancePair(nums, k)); // 输出 0
}
}
代码说明
排序数组:
首先对数组进行排序,这样可以方便地使用双指针方法来计算绝对差值。
二分查找:
使用二分查找确定最终的差值阈值。初始时,left设为0,right设为数组中最大值与最小值之差。
计算小于等于阈值的差值对数量:
countPairs方法使用双指针来计算数组中差值小于等于mid的对数。如果这种对数少于k,说明第k小的差值肯定比mid大,所以调整二分查找的范围。
返回结果:
最终,left会收敛到第k小的差值。
测试示例
在main方法中提供了一个简单的测试用例,数组{1, 3, 1}和k = 1,结果应该输出0,因为最小的差值对是(1, 1),其差值为0。
要在一个数组中找到所有两两差值绝对值中的第M小的数,可以通过二分查找的方法来实现。这个方法的思路是:通过二分查找确定一个差值阈值,然后利用双指针或者其他方法计算数组中与这个阈值相等或更小的差值的个数,从而判断这个阈值是否是第M小的差值。
以下是Java实现的代码:
java
复制
import java.util.Arrays;
public class KthSmallestPairDistance {
public static int smallestDistancePair(int[] nums, int k) {
Arrays.sort(nums);
int left = 0;
int right = nums[nums.length - 1] - nums[0];
while (left < right) {
int mid = left + (right - left) / 2;
if (countPairs(nums, mid) < k) {
left = mid + 1;
} else {
right = mid;
}
}
return left;
}
private static int countPairs(int[] nums, int mid) {
int count = 0;
int left = 0;
for (int right = 0; right < nums.length; right++) {
while (nums[right] - nums[left] > mid) {
left++;
}
count += right - left;
}
return count;
}
public static void main(String[] args) {
int[] nums = {1, 3, 1};
int k = 1;
System.out.println(smallestDistancePair(nums, k)); // 输出 0
}
}
代码说明
排序数组:
首先对数组进行排序,这样可以方便地使用双指针方法来计算绝对差值。
二分查找:
使用二分查找确定最终的差值阈值。初始时,left设为0,right设为数组中最大值与最小值之差。
计算小于等于阈值的差值对数量:
countPairs方法使用双指针来计算数组中差值小于等于mid的对数。如果这种对数少于k,说明第k小的差值肯定比mid大,所以调整二分查找的范围。
返回结果:
最终,left会收敛到第k小的差值。
测试示例
在main方法中提供了一个简单的测试用例,数组{1, 3, 1}和k = 1,结果应该输出0,因为最小的差值对是(1, 1),其差值为0。
全部评论
相关推荐
10-24 23:06
华中科技大学附属小学 Java 点赞 评论 收藏
分享
10-10 17:25
西安电子科技大学 C++ 点赞 评论 收藏
分享