- 1. 题目链接
https://leetcode-cn.com/problems/intersection-of-two-arrays-ii/
- 2. 题目描述
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明:
输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
- 3. 题目分析
根据题目要求,两个数组的交集,可以不考虑输出结果的顺序,那么可以先用qsort函数对两个数组排序,然后从小到大比较nums1[i] 和 nums2[j],较小的下标加一,相等时,下标同时加一。
- 4. 代码实现
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
#define max(a, b) (a > b ? a : b)
int comp(const void *a, const void *b) {
if(*(int*)a < *(int*)b)
return -1;
else if( *(int*)a == *(int*)b)
return 0;
else
return 1;
}
int* intersect(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize){
if (nums1Size == 0 || nums2Size == 0) {
*returnSize = 0;
return NULL;
}
int cnt = 0;
int i = 0, j = 0;
int *res = (int*)malloc(sizeof(int) * max(nums1Size, nums2Size));
qsort(nums1, nums1Size, sizeof(int), comp);
qsort(nums2, nums2Size, sizeof(int), comp);
while (i < nums1Size && j < nums2Size) {
if (nums1[i] < nums2[j]) {
i++;
continue;
}
if (nums1[i] > nums2[j]) {
j++;
continue;
}
if (nums1[i] == nums2[j]) {
res[cnt++] = nums1[i];
i++;
j++;
}
}
*returnSize = cnt;
return res;
}
- 5. 调试报错
在执行用例
[-2147483648,1,2,3]
[1,-2147483648,-2147483648]
时报错,如下图所示,
发现comp函数定义中,返回值执行(-2147483648 - 1)操作会超出int型范围。
int comp(const void *a, const void *b) {
return *(int *)a - *(int *)b;
}
- 6. 提交记录