力扣LeetCode，两个数组的交集

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

示例 1:

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

示例 2:

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

说明:

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

2、解题思路和代码如下所示：

package com.leetcode;


import java.util.*;

/**
 * @ProjectName: dataConstruct
 * @Package: com.leetcode
 * @ClassName: ArrayIntersection
 * @Author: biehl
 * @Description: ${description}
 * @Date: 2020/3/15 10:55
 * @Version: 1.0
 */
public class ArrayIntersection {

    /**
     * 给定两个数组，编写一个函数来计算它们的交集。
     *
     * @param nums1
     * @param nums2
     * @return
     */
    public int[] intersection(int[] nums1, int[] nums2) {
        // 将数组nums1的元素数据放入到集合Set中
        Set<Integer> setNums1 = new TreeSet<Integer>();
//        for (int i = 0; i < nums1.length; i++) {
//            setNums1.add(nums1[i]);
//        }
        for (int num : nums1) {
            setNums1.add(num);
        }


        // 将数组nums2的元素数据放入到集合Set中
        Set<Integer> setNums2 = new TreeSet<Integer>();
//        for (int i = 0; i < nums2.length; i++) {
//            setNums2.add(nums2[i]);
//        }
        for (int num : nums2) {
            setNums2.add(num);
        }

        List<Integer> result = new ArrayList<Integer>();
        // 开始遍历，查看第一个集合中的元素在第二个集合中有多少个
        Iterator<Integer> iterator = setNums1.iterator();
        while (iterator.hasNext()) {
            Integer next = iterator.next();
            // 如果第一个集合中的元素在第二个集合中，就进行保存操作
            if (setNums2.contains(next)) {
                result.add(next);
            }
        }

        int[] res = new int[result.size()];
        for (int i = 0; i < result.size(); i++) {
            res[i] = result.get(i);
        }
        return res;
    }


    /**
     * 给定两个数组，编写一个函数来计算它们的交集。
     *
     * @param nums1
     * @param nums2
     * @return
     */
    public int[] intersection2(int[] nums1, int[] nums2) {
        // 将数组nums1的元素数据放入到集合Set中
        Set<Integer> setNums1 = new TreeSet<Integer>();
        for (int num : nums1) {
            setNums1.add(num);
        }

        // 声明一个动态数组，存放将要返回的数据
        List<Integer> result = new ArrayList<Integer>();
        for (int num : nums2) {
            // 如果包含了这个元素
            if (setNums1.contains(num)) {
                // 将这个交集元素存储到动态数组中
                result.add(num);
                // 为了避免出现重复数据，可以将setNums1集合中重复的数据删除掉。下次就不会出现重复的数据了。
                setNums1.remove(num);
            }
        }

        // 将动态数据中的数据，封装返回即可
        int[] res = new int[result.size()];
        for (int i = 0; i < result.size(); i++) {
            res[i] = result.get(i);
        }
        return res;
    }

    public static void main(String[] args) {
//        int[] nums1 = new int[]{1, 2, 2, 1};
//        int[] nums2 = {2, 2};
        int[] nums1 = new int[]{4, 9, 5};
        int[] nums2 = {9, 4, 9, 8, 4};
        ArrayIntersection arrayIntersection = new ArrayIntersection();
        int[] intersection = arrayIntersection.intersection2(nums1, nums2);
        for (int i = 0; i < intersection.length; i++) {
            System.out.println(intersection[i]);
        }
    }

}