[LeetCode] 349.两个数组的交集（Easy）C语言题解

题目

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

---

示例

①示例1

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

②示例2

• 输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
• 输出: [9,4]

---

说明

输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。

①数据范围(自测)

• 0 <= nums1[i] < 1000
• 0 <= nums2[i] < 1000

②相关话题

• 排序
• 哈希表
• 双指针
• 二分查找

③相似题目

• 350. 两个数组的交集 II — 力扣网
• 350. Intersection of Two Arrays II — leetcode

④题目地址

• 349. 两个数组的交集 — 力扣网
• 349. Intersection of Two Arrays — leetcode

---

解题方法

①哈希表

• 数据的范围大概是 0 ~ 1000，可用数组模拟的哈希表求解，速度最快(用空间换时间)。
• 由于问题中的元素是唯一的，所以我们只关心元素的有无(用哈希表记录元素是否出现过)。
• 用数组模拟的哈希表求解虽然直观，但是会浪费一些内存空间，所以更好的方法是使用 uthash。
• uthash 是一个用 C 语言编写的开源库，使用宏实现了哈希表的增删改查等功能。
• 时间复杂度：O(N)。
• 空间复杂度：O(N)。

②排序

• 数组模拟的哈希表只适用于数据范围一定且数据浮动较小的情况，而使用 排序+双指针查找 的方法更灵活，可以先使用快排函数排序两个数组，再通过双指针查找交集。
• 时间复杂度：O(Nlogn)。
• 空间复杂度：O(1)。

---

代码详解

• 哈希表

int* intersection(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize) {
    // 哈希表。
    int hash[1000] = {0};
    int* result = malloc(sizeof(int)*nums1Size);
    *returnSize = 0;

    for (int i = 0; i < nums1Size; i++)
        hash[nums1[i]] = 1;
    for (int i = 0; i < nums2Size; i++) {
        // 查找重复数字。
        if (hash[nums2[i]] == 1) {
            result[(*returnSize)++] = nums2[i];
            // 去重。
            hash[nums2[i]] = 0;
        }
    }
    
    return result;
}

• uthash

int* intersection(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize) {
    struct hash {
        int value;
        // makes this structure hashable.
        UT_hash_handle hh; 
    };
    
    struct hash *hashTable = NULL;
    int* result = malloc(sizeof(int)*nums1Size);
    
    for (int i = 0; i < nums1Size; i++) {
        struct hash *h;
        // 查找值。
        HASH_FIND_INT(hashTable, nums1+i, h);
        
        if (!h) {
            h = malloc(sizeof(struct hash));
            h->value = nums1[i];
            // 若值不存在则添加。
            HASH_ADD_INT(hashTable, value, h);
        }
    }
    for (int i = 0; i < nums2Size; i++) {
        struct hash *h;
        HASH_FIND_INT(hashTable, nums2+i, h);
        
        // 找到交集元素后，得从哈希表中删除它(因为输出结果中的每个元素是唯一的)。
        if (h) {
            result[(*returnSize)++] = h->value;
            HASH_DEL(hashTable, h);
        }
    }
    
    return result;
}

• 排序 + 双指针查找

// 快排函数(升序)。
int compare(const void* a, const void* b) {
    return (*(int*)a)-(*(int*)b); 
}

int* intersection(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize) {    
    // 排序。
    qsort(nums1, nums1Size, sizeof(int), compare);
    qsort(nums2, nums2Size, sizeof(int), compare);
    
    int pos1 = 0, pos2 = 0; 
    int* result = malloc(sizeof(int)*nums1Size);
    *returnSize = 0;
    
    // 双指针。
    while (pos1 < nums1Size && pos2 < nums2Size) {
        if (nums1[pos1] < nums2[pos2])
            pos1++;
        else if (nums1[pos1] > nums2[pos2])
            pos2++;
        else {
            // 去重。
            if (*returnSize == 0 || nums1[pos1] != result[*returnSize-1])
                result[(*returnSize)++] = nums1[pos1];
            
            pos1++;
            pos2++;
        }
    }
    
    return result;
}

  | ps：此代码删去去重部分，就是 350. 两个数组的交集 II — 力扣网 的题解。

---

附录

• 我的个人博客：messi1002.top
• 如有错误或疑惑之处 请联系 wjymessi@163.com
• 所有题目解答：fork me on github