Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情。
一、题目描述:
349. 两个数组的交集 - 力扣(LeetCode) (leetcode-cn.com)
给定两个数组 nums1 和 nums2 ,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
解释:[4,9] 也是可通过的
提示:
- 1 <= nums1.length, nums2.length <= 1000
- 0 <= nums1[i], nums2[i] <= 1000
二、思路分析:
可以用到集合来解决,集合一般是没有重复元素的,那么就可以先对数组1去重,去重后再遍历数组2,如果在集合中存在的,说明是数组1和数组2都有的元素,就加入结果数组中,为了避免数组2中重复这个元素,需要把集合中的删除,第二次再遇到时集合里面没有了,就不会重复添加到结果数组中。
三、AC 代码:
import java.util.ArrayList;
import java.util.TreeSet;
public class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
TreeSet<Integer> set = new TreeSet<>();
for (int i:nums1) {
set.add(i);
}
ArrayList<Integer> arrayList = new ArrayList<>();
for (int i:nums2) {
if(set.contains(i)){
arrayList.add(i);
set.remove(i);
}
}
int[] res = new int[arrayList.size()];
for (int i=0;i<arrayList.size();i++){
res[i] = arrayList.get(i);
}
return res;
}
}
四、总结:
写完开开心心过周末了~
范文参考: