C刷题:LeetCode 373. 查找和最小的K对数字 || 华为面试题
作者:来知晓
公众号:来知晓
刷题交流QQ群:444172041
LeetCode 373. 查找和最小的K对数字,题目描述,戳此处。
核心思路
感谢LeetCode题解中cnwsssss提供的实现思路,虽然ta用的是JavaScript代码。
思路如下:
- 不用优先队列/堆等复杂结构,就用双指针思路
- 按升序排序后,做一个for循环遍历最次找最小的值,排列组合即可
- 新申请一个数组
steps[]
,记录nums1[i]
与nums2[]
数组结合的起始位置 - 每次找1个最小值,就在
nums1
中for循环匹配一下最小的,从steps[i]
的起点开始 - 找到最小值后,
steps[n1minIdx]
的值加1
代码实现
int** kSmallestPairs(int* nums1, int nums1Size, int* nums2, int nums2Size, int k, int* returnSize, int** returnColumnSizes)
{
long long tmp = (long long)nums1Size * nums2Size;
k = (k > tmp) ? tmp : k;
int *steps = (int*)malloc(nums1Size * sizeof(int));
if (steps == NULL) {
return NULL;
}
memset(steps, 0, nums1Size * sizeof(int));
// 申请返回内存
int **returnArr = (int**)malloc(k * sizeof(int*) + 100); // 返回每行的列数
if (returnArr == NULL) {
return NULL;
}
*returnColumnSizes = (int*)malloc(k * sizeof(int) + 100); // 返回每行的列数
if (*returnColumnSizes == NULL) {
return NULL;
}
int n1MinIdx = 0;
int i;
for (i = 0; i < k; i++) {
int min = INT_MAX; // min要放在每一轮查找前初始,以免成了上一轮的min
for (int j = 0; j < nums1Size; j++) { // 找一次两两匹配的最小值
if (steps[j] < nums2Size && nums1[j] + nums2[steps[j]] < min) {
min = nums1[j] + nums2[steps[j]];
n1MinIdx = j;
}
}
// 最终得到最小值对应的nums1的idx,以及对应的nums2[]下标steps[idx]
int *p = (int*)malloc(2 * sizeof(int));
if (p == NULL) {
return NULL;
}
p[0] = nums1[n1MinIdx];
p[1] = nums2[steps[n1MinIdx]];
steps[n1MinIdx] += 1; // n1MindIdx对应nums2下标值后移一位
returnArr[i] = p;
returnColumnSizes[0][i] = 2; // 每行的列数为2
}
*returnSize = k;
return returnArr;
}
JavaScript代码参考
/*
参考思路:
和之前几个题目思想很像,用一个数组来记录num1中每个元素在nums2中走了多远就可以,每次循环都是nums1和nums2加起来最小的nums1往前走一步
*/
var kSmallestPairs = function(nums1, nums2, k) {
if (k > nums1.length * nums2.length ) {
k = nums1.length * nums2.length
}
if (nums1.length == 0 || nums2.length == 0) {
return [];
}
let steps = new Array(nums1.length);
for (let i = 0; i < steps.length; i++) {
steps[i] = 0;
}
let results = [];
for (let i = 0; i < k; i++) {
let min = Number.MAX_VALUE;
let minStepIndex = 0;
for (let j = 0; j < nums1.length; j++) {
if (steps[j] < nums2.length && nums2[steps[j]] + nums1[j] < min) {
min = nums2[steps[j]] + nums1[j];
minStepIndex = j;
}
}
results.push([nums1[minStepIndex], nums2[steps[minStepIndex]]]);
steps[minStepIndex]++;
}
return results;
};
引申问题
- 思考下本题的复杂版本,实现一个优先队列数据结构来得到前k个最小对?
- 最小队列的时间复杂度可能比以上实现低,但代价是增加了内存。