leetcode刷题记录之349

349、两个数组的交集

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

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

示例 2：

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

说明：

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

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

1、C语言代码：

//解法一
/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int *intersection(int *nums1, int nums1Size, int *nums2, int nums2Size, int *returnSize){
	int a[1000] = { 0 }, b[1000] = { 0 }, *c = (int*)malloc(sizeof(int)*nums1Size), i, max;
 	for (i = max = 0; i < nums1Size; i++){    //建立两哈希表并寻找出最大值 max
  		if (max < nums1[i]) max = nums1[i];
  		a[nums1[i]] = 1;
 	}
 	for (i = 0; i < nums2Size; i++){
  		if (max < nums2[i]) max = nums2[i];
  		b[nums2[i]] = 1;
 	}
 	for (i = *returnSize = 0; i <= max; i++)   //遍历哈希表寻找交集
  		if (a[i] && b[i]) c[(*returnSize)++] = i;
 	return c;
}

//解法二
int* intersection(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize){
    int mark[17000]={0};
    for(int i=0;i<nums1Size;i++){
        mark[nums1[i]]++;        
    }
    int maxsize=nums1Size>nums2Size?nums2Size:nums1Size;
    int *c=(int *)malloc(sizeof(int)*maxsize);
    int k=0;
    for(int j=0;j<nums2Size;j++){
        if(mark[nums2[j]]>0){
            c[k++]=nums2[j];          
            mark[nums2[j]]=0;
        }
    }
    *returnSize=k;
    return c;
}

解释： 解法一：建立两个哈希表，每个哈希表用很大的数组（初始值为 0）表示，遍历数组，用数组元素的值当作哈希表的下标，赋值为 1，这样两个哈希表就用值为 1 的数组元素的下标表示了两数组中存在的数字，遍历两哈希表，在相同下标下值都为 1 的元素即二者的交集，下标便是交集的值，将其存到返回数组中即可。
解法二：计数标记。

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

2、Java代码：

class Solution {	
     public int[] set_intersection(HashSet<Integer> set1, HashSet<Integer> set2) {
    	int [] output = new int[set1.size()];
    	int idx = 0;
    	for (Integer s : set1)
      	    if (set2.contains(s)) output[idx++] = s;
    	        return Arrays.copyOf(output, idx);
  	    }
     public int[] intersection(int[] nums1, int[] nums2) {
    	HashSet<Integer> set1 = new HashSet<Integer>();
    	for (Integer n : nums1) set1.add(n);
    	HashSet<Integer> set2 = new HashSet<Integer>();
    	for (Integer n : nums2) set2.add(n);
    	if (set1.size() < set2.size()) return set_intersection(set1, set2);
    	else return set_intersection(set2, set1);
  	}
}

解释： 其思想是将两个数组转换为集合 set，然后迭代较小的集合检查是否存在在较大集合中。

知识点回顾： 无。

3、Python代码：

class Solution:
    def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
        return set(nums1) & set(nums2)

解释： 利用set。

知识点回顾： 集合（set）是一个无序的不重复元素序列。

4、JavaScript代码：

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersection = function(nums1, nums2) {
    return [...new Set(nums1)].filter( item => nums2.includes(item) )
};

解释： 无。

知识点回顾： 无。