LeetCode 349. Intersection of Two Arrays

349. Intersection of Two Arrays

Description

Given two arrays, write a function to compute their intersection.

Example:Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2].Note:1.Each element in the result must be unique.2.The result can be in any order.

Solution

• 题意即求两个数组的交集
• 我的做法并不好，如果数据范围太大就会造成溢出。
• 我的做法是用两个桶，将每个数组对应的数用0-1表示，比如nums1中有个数是6，那么num1（桶）下标为6的位置用1填充，最后两个同做与运算，从而得到交集，代码如下

int* intersection(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize) {    int *rnt = (int *)malloc(sizeof(int) * (nums1Size > nums2Size ? nums1Size : nums2Size));    memset(rnt,0,sizeof(rnt));    if (nums1Size == 0 || nums2Size == 0) return rnt;    int num1[100010];    int num2[100010];    memset(num1,0,sizeof(num1));    memset(num2,0,sizeof(num2));    int max1 = nums1[0],max2 = nums2[0];    for (int i = 0;i < nums1Size;i++) {        if (nums1[i] > max1) max1 = nums1[i];        num1[nums1[i]] = 1;    }    for (int i = 0;i < nums2Size;i++) {        if (nums2[i] > max2) max2 = nums2[i];        num2[nums2[i]] = 1;    }    int MAX = max1 > max2 ? max1 : max2;    int count = 0;    for (int i = 0;i <= MAX;i++) {        if (num1[i] && num2[i]) {            rnt[count++] = i;        }    }    *returnSize = count;    return rnt;}

• 下面给出一个set的做法，由题解给出，侵删。

vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {    set<int> s(nums1.begin(), nums1.end());    vector<int> out;    for (int x : nums2)        if (s.erase(x))            out.push_back(x);    return out;}