LeetCode 349 两个数组的交集

1. 题目链接

https://leetcode-cn.com/problems/intersection-of-two-arrays/

2. 题目描述

  给定两个数组，编写一个函数来计算它们的交集。

示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]

示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]

说明:

输出结果中的每个元素一定是唯一的。

我们可以不考虑输出结果的顺序。

3. 题目解析

根据题目要求，两个数组的交集，可以不考虑输出结果的顺序，那么可以先用qsort函数对两个数组排序，然后从小到大比较nums1[i] 和 nums2[j]，较小的下标加一，相等时，下标同时加一。因为结果中的每个元素是唯一的，增加去重判断，即 nums1[i] == nums1[i + 1] 时跳过。

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) {
    return *(int *)a - *(int *)b;
}

int* intersection(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 < nums1Size - 1 && nums1[i] == nums1[i + 1])) {
            i++;
            continue;
        }
        if (nums1[i] > nums2[j] || (j < nums2Size - 1 && nums2[j] == nums2[j + 1])) {
            j++;
            continue;
        }
        if (nums1[i] == nums2[j]) {
            res[cnt++] = nums1[i];
            i++;
            j++;
        }
    }
    *returnSize = cnt;
    return res;
}

5. 提交记录