LeetCode 349. 两个数组的交集【Java】

1.题目

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

示例 1：
输入：nums1 = [1,2,2,1], nums2 = [2,2]
输出：[2]

示例 2：
输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]
 
说明：
输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。
通过次数220,713提交次数299,025

2.思路和代码

先排序，如果两个数组的值相同并且不和返回数组内的最后一个元素重复，则将此元素打入数组，直到遍历完两个数组，最后返回前k个元素

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        int[] sum = new int[nums1.length + nums2.length];//返回数组
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        int i = 0;//nums1的下标
        int j = 0;//nums2的下标
        int k = 0;//记录两个数组元素值相等的个数
        while(i != nums1.length && j != nums2.length){
            if(nums1[i] == nums2[j] ){
                if(k == 0 || sum[k - 1] != nums1[i]){
                    sum[k++] = nums1[i];
                }
                i++;
                j++;
            }
            else if(nums1[i] < nums2[j]){
                i++;
            }
            else{
                j++;
            }
        }
        return Arrays.copyOfRange(sum, 0, k);//返回前k个元素 k+1之后的元素均为初始化的0
    }
}