349. 两个数组的交集

2020-01-19  本文已影响0人  周英杰Anita

题目描述:

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

示例 1:

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

示例 2:

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

说明:

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

思路:

1. 首先排序两个数组
2. 从前向后依次遍历数组
3. 两个数组元素相等,判断结果数组中是否已有该元素,没有就添加,有就i++,j++
4. 如果nums1[i] < nums2[j]则 i++;如果nums1[i] > nums2[j] 则 j++;

Java解法:

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        int len1 = nums1.length;
        int len2 = nums2.length;
        List<Integer> ans = new ArrayList<Integer>();
        for(int i = 0, j = 0; i < len1 && j <len2;)
        {
            if(nums1[i] < nums2[j]){
                i++;
            }else if(nums1[i] > nums2[j]){
                j++;
            }else{
                if(!ans.contains(nums1[i])){
                    ans.add(nums1[i]);
                }
                i++;
                j++;
            }
        }
        int[] answer = new int[ans.size()];
        for (int i = 0; i < ans.size(); i++) {
            answer[i] = ans.get(i);
        }
        return answer;
    }
}

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/intersection-of-two-arrays/

上一篇 下一篇

猜你喜欢

热点阅读