LeetCode 350 两个数组的交集II

• 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. 提交记录