1. 程式人生 > >LeetCode——陣列:兩個陣列的交集 II

LeetCode——陣列:兩個陣列的交集 II

原題

給定兩個陣列,寫一個方法來計算它們的交集。

例如:
給定 nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2, 2].

注意:

  •    輸出結果中每個元素出現的次數,應與元素在兩個陣列中出現的次數一致。
  •    我們可以不考慮輸出結果的順序。

跟進:

  • 如果給定的陣列已經排好序呢?你將如何優化你的演算法?
  • 如果 nums1 的大小比 nums2 小很多,哪種方法更優?
  • 如果nums2的元素儲存在磁碟上,記憶體是有限的,你不能一次載入所有的元素到記憶體中,你該怎麼辦


個人程式設計水平不高,沒有做跟進的題目。

程式碼如下:

public int[] intersect(int[] nums1, int[] nums2) {
		 
		/*
	        思路使用兩個HashMap來輔助,將nums1中的元素與nums2中的元素比較,相等的話分別將陣列的索引和對應的值新增到hs3和hs4中,
	        因為要避免重複,需要在if()裡面判斷這兩HashMap中是否已存有這兩陣列的鍵值
	      */
		HashMap<Integer,Integer> hs3 = new HashMap<Integer,Integer>();
		HashMap<Integer,Integer> hs4 = new HashMap<Integer,Integer>();
		

		for(int i = 0; i < nums1.length; i++) {
			for(int j = 0; j < nums2.length; j++) {
				if(nums1[i] == nums2[j] && !hs3.containsKey(i) && !hs4.containsKey(j)) {
					hs3.put(i, nums1[i]);
					hs4.put(j, nums2[j]);
				}
			}
		}
		
		ArrayList<Integer> al = new ArrayList<Integer>(hs3.values());
		int[] ts = new int[al.size()];
		for(int i = 0; i < al.size(); i++) {
			ts[i] = al.get(i);
		}
		
		return ts;
		
	 }
提供一種解題思路,也方便日後回顧和改進。