版权声明:转载请注明出处: https://blog.csdn.net/qq_34774655/article/details/85337982
运用的知识: 映射 。 参考:自定义映射(Map)
题目描述:
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2] 输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出: [4,9]
说明:
- 输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
- 我们可以不考虑输出结果的顺序。
解答:
import java.util.ArrayList;
import java.util.TreeMap;
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
//由于题目要求输出结果唯一,所以想到用映射
//第一个代表主键,第二个代表出现的次数
TreeMap<Integer,Integer> treeMap=new TreeMap<Integer,Integer>();
//遍历数组1,添加到映射中
for(int num:nums1) {
if(!treeMap.containsKey(num)) {
treeMap.put(num, 1);
}else {
treeMap.put(num, treeMap.get(num)+1);
}
}
ArrayList<Integer> list=new ArrayList<>();
//遍历数组2,将交集元素,放入list中
for(int num:nums2) {
if(treeMap.containsKey(num)) {
list.add(num);
treeMap.replace(num, treeMap.get(num)-1);
if(treeMap.get(num)==0) {
treeMap.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;
}
}