LeetCode——陣列:兩個陣列的交集 II
阿新 • • 發佈:2018-12-27
原題
給定兩個陣列,寫一個方法來計算它們的交集。
例如:
給定 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; }