LeetCode之349. 两个数组的交集

华der

于 2018-12-29 13:53:30 发布

阅读量116

点赞数

分类专栏： LeetCode 文章标签： LeetCode

本文链接：https://blog.youkuaiyun.com/qq_34774655/article/details/85337720

版权

LeetCode 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种使用Java实现的算法，该算法可以计算并返回两个整数数组的交集，确保结果中每个元素都是唯一的。通过使用TreeSet去除重复元素，并遍历第二个数组检查交集，最终将结果转换为int数组返回。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

运用的知识：集合。参考：自定义集合（Set）

题目描述：

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

示例 1:

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

示例 2:

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

说明:

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

解答：



import java.util.ArrayList;
import java.util.TreeSet;

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
    	//由于题目要求输出结果唯一，所以想到用集合
        TreeSet<Integer> set=new TreeSet<Integer>();
        //将数组1中的元素去重后，添加到set集合
        for(int num:nums1) {
        	set.add(num);
        }
        ArrayList<Integer> list=new ArrayList<Integer>();
        //遍历数组2，将交集元素，放入list中
        for(int num:nums2) {
        	if(set.contains(num)) {
        		list.add(num);
        		set.remove(num);
        	}
        }
        //将list集合，变成int数组返回
        int[] result=new int[list.size()];
        for(int i=0;i<list.size();i++) {
        	result[i]=list.get(i);
        	
        }
        return result;
 
    }
}