leetcode刷题记录之350

350、两个数组的交集 II

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

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

示例 2：

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

说明：

• 输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
• 我们可以不考虑输出结果的顺序。

进阶：

• 如果给定的数组已经排好序呢？你将如何优化你的算法？
• 如果 nums1 的大小比 nums2 小很多，哪种方法更优？
• 如果 nums2 的元素存储在磁盘上，磁盘内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？

难度：简单  题目地址：https://leetcode-cn.com/problems/intersection-of-two-arrays-ii/

1、C语言代码：

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int cmp(const void* a, const void* b){
    return *(int*)a > *(int*)b;
}
int* intersect(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize){
    *returnSize = nums1Size > nums2Size ? nums2Size : nums1Size;
    int *ans = (int*)malloc(sizeof(int) * (*returnSize));
    qsort(nums1, nums1Size, sizeof(int), cmp);
    qsort(nums2, nums2Size, sizeof(int), cmp);
    int i = 0, j = 0, k = 0;
    while(i < nums1Size && j < nums2Size){
        if(nums1[i] > nums2[j])
            j++;
        else if(nums1[i] < nums2[j])
            i++;
        else{
            ans[k++] = nums1[i];
            i++; j++;
        }
    }
    *returnSize = k;
    return ans;
}

解释： 先排序，然后依次对比2个数组中的元素，相等就放入新创立的数组中。

知识点回顾： 无。

2、Java代码：

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        HashMap<Integer, Integer> map = new HashMap<>();
        List<Integer> list = new ArrayList<>();
        for(int num : nums1) {
            if(!map.containsKey(num)) map.put(num, 1);
            else map.put(num, map.get(num) + 1);
        }
        for(int num : nums2) {
            if(map.containsKey(num)) {
                map.put(num, map.get(num) - 1);
                if(map.get(num) == 0) map.remove(num);
                list.add(num);
            }
        }
        int[] res = new int[list.size()];
        for(int i = 0; i < list.size(); i++) {
            res[i] = list.get(i);
        }
        return res;
    }
}

解释： 使用哈希表记录数组1中的数和其对应频次。遍历数组2，如果当前索引的数在哈希表中存在，则把它丢进结果list中，并把它在哈希表中的频次-1，如果频次为0就移除它。完成遍历后，list中就是我们希望得到的结果。

知识点回顾： 哈希表的相关知识，将在数据结构篇中体现。

3、Python代码：

class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
        return [nums2.pop(nums2.index(n)) for n in nums1 if n in nums2]

解释： 无。

知识点回顾： 无。

4、JavaScript代码：

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersect = function(nums1, nums2) {
    return nums1.filter(el => nums2.includes(el) && nums2.splice(nums2.indexOf(el), 1))    
};

解释： 无。

知识点回顾： 无。