前端與算法 leetcode 350. 兩個數組的交集 II

  • 2019 年 11 月 10 日
  • 筆記

# 前端與算法 leetcode 350. 兩個數組的交集 II

題目描述

給定兩個數組,編寫一個函數來計算它們的交集。

示例 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 的元素存儲在磁盤上,磁盤內存是有限的,並且你不能一次加載所有的元素到內存中,你該怎麼辦?

350. 兩個數組的交集 II

概要

這道題可以看成一道傳統的映射題,我們需要知道每個值出現的次數,映射關係為<元素,出現次數>,首先統計數組1中所有元素出現的次數,再遍曆數組2即可,如果數組2中的元素存在且大於1,則表示已經找到一個相同的了,如果值等於1,直接刪除[1]

提示

哈希表,雙指針,暴力

解析

解法一:哈希表

時間複雜度O(n)
先用Hashmap記錄第一個數組中的元素【放在key】,和出現的次數【放在value】。然後再遍歷第二個數組,如果找到對應元素,則添加這個元素到返回數組裡。如果value值大於1,HashMap中的value值減 1,表示已經找到一個相同的了。如果value值等於1,則刪除該元素。[2]

解法二:雙指針

將兩個數組排序,兩個指針初始值為0,比較兩個指針的元素是否相等相等則放到返回數組裡,兩個指針同時往前,不相等元素小的指針往前,如果相等,那肯定比較過的元素就沒用了,兩個指針往前,不相等的時候因為較大的元素在較小的數組裡存在所以把元素小的數組指針往前

解法三:暴力法

時間複雜度O(n^2)
遍歷第一個數組,然後再第二個數組查找indexOf是否有當前元素

算法

/**   * @param {number[]} nums1   * @param {number[]} nums2   * @return {number[]}   */  var intersect = function (nums1, nums2) {  // hash法    const [hash,res] = [new Map(),[]]    nums1.forEach(el=>{      if (hash.has(el)) {        hash.set(el, hash.get(el) + 1)      } else {        hash.set(el, 1)      }    })    nums2.forEach(el=>{      const hashKey = hash.get(el)      if (hash.has(el)) {        res.push(el)        if (hashKey > 1) {          hash.set(el, hashKey - 1)        } else {          hash.delete(el)        }      }    })    return res  //   // 雙指針法  //   let p1 = 0  //   let p2 = 0  //   let res = []  //   nums1 = nums1.sort((a, b) => a - b)  //   nums2 = nums2.sort((a, b) => a - b)  //   while (p1 < nums1.length && p2 < nums2.length) {  //     if (nums1[p1] == nums2[p2]) {  //       res.push(nums1[p1])  //       p1++  //       p2++  //     } else if (nums1[p1] < nums2[p2]) {  //       p1++  //     } else {  //       p2++  //     }  //   }  //   return res  //   // 暴力法  // const res = []  // if (nums1.length < nums2.length) [nums1, nums2] = [nums2, nums1]  // for (let i = 0; i < nums1.length; i++) {  //   const key = nums2.indexOf(nums1[i])  //   if (key !== -1) res.push(nums2.splice(key, 1))  // }  // return res  }

傳入[1, 2], [11, 1, 2, 3, 2]的運行結果

[ 1, 2 ]

執行結果

執行用時 :56 ms, 在所有 javascript 提交中擊敗了99.55% 的用戶  內存消耗 :34.7 MB, 在所有 javascript 提交中擊敗了53.72%的用戶