LeetCode之350. 两个数组的交集 II

运用的知识：  映射    。              参考：自定义映射（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;
 
		 
	    }
}