題目描述
給定兩個數組，編寫一個函數來計算它們的交集。

示例 1:

輸入: nums1 = [1,2,2,1], nums2 = [2,2] 輸出: [2,2]

示例 2:

輸入: nums1 = [4,9,5], nums2 = [9,4,9,8,4] 輸出: [4,9]

說明：

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

進階:

• 如果給定的數組已經排好序呢？你將如何優化你的算法？
• 如果 nums1 的大小比 nums2 小很多，哪種方法更優？
• 如果 nums2 的元素存儲在磁盤上，磁盤內存是有限的，并且你不能一次加載所有的元素到內存中，你該怎么辦？

思路：先進行排序，然后遍歷獲取交集即可
代碼如下：

public int[] intersect(int[] nums1, int[] nums2) {Arrays.sort(nums1);Arrays.sort(nums2);return intersect1(nums1.length>=nums2.length?nums1:nums2, nums1.length>=nums2.length?nums2:nums1);}private int[] intersect1(int[] longArr, int[] shortArr) {int[] res = new int[shortArr.length];int count=0, j=0;for(int i=0;i<shortArr.length;i++) {for(;j<longArr.length;j++) {if(shortArr[i] == longArr[j]) {res[count++]=longArr[j++];break;}else if (shortArr[i] < longArr[j]) {break;}}}return Arrays.copyOf(res, count);}

改進一下，使用單層循環

public int[] intersect(int[] nums1, int[] nums2) {Arrays.sort(nums1);Arrays.sort(nums2);return intersect1(nums1.length>=nums2.length?nums1:nums2, nums1.length>=nums2.length?nums2:nums1);}private int[] intersect1(int[] longArr, int[] shortArr) {int[] res = new int[shortArr.length];int count=0;for(int i=0, j=0;i<shortArr.length && j<longArr.length;i++) {if (shortArr[i] > longArr[j]) {while (shortArr[i] > longArr[j] && j<longArr.length-1) {j++;}}if(shortArr[i] == longArr[j]) {res[count++]=longArr[j++];}}return Arrays.copyOf(res, count);}

總結

以上是生活随笔為你收集整理的2019-03-14-算法-进化（两个数组的交集 II）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。