LeetCode

两个数组的交集

2019-06-17  本文已影响0人  习惯了_就好

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

示例 1:

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

示例 2:

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

说明:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/intersection-of-two-arrays
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        int length1 = nums1.length;
        int length2 = nums2.length;
        
        Set<Integer> set1 = new HashSet();
        Set<Integer> set2 = new HashSet();
        
        for(int i = 0; i < length1; i++){
            set1.add(nums1[i]);
        }
        
        for(int i = 0; i < length2; i++){
            set2.add(nums2[i]);
        }
        
        int[] result = new int[set1.size()];
        
        int i = 0;
        for(int temp :set2){
          if(set1.contains(temp)){
              result[i] = temp;
              i++;
          } 
        }
        
        return Arrays.copyOfRange(result, 0, i);
    }
}
上一篇 下一篇

猜你喜欢

热点阅读