代码随想录day7 - 哈希表部分(下)

91 阅读3分钟

454.四数相加II

力扣题目链接(opens new window)

给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ,使得 A[i] + B[j] + C[k] + D[l] = 0。

为了使问题简单化,所有的 A, B, C, D 具有相同的长度 N,且 0 ≤ N ≤ 500 。所有整数的范围在 -2^28 到 2^28 - 1 之间,最终结果不会超过 2^31 - 1 。

例如:

输入:

  • A = [ 1, 2]
  • B = [-2,-1]
  • C = [-1, 2]
  • D = [ 0, 2]

输出: 2

解题思路

四数相加 可以看做两数之和的加倍版本,而且这个输出只需要求解的个数,所以可以把四个数组分为两组,分别求取前两组的和的值,放入map中,然后遍历后两组取和,当 0-后两组的和 的值在map中时,结果加上map.get(xxx)注意,不是+1

var fourSumCount = function(nums1, nums2, nums3, nums4) {
    let mapAB = new Map()
    for(let num1 of nums1){
        for(let num2 of nums2){
            let sum = num1 + num2
            if(!mapAB.has(sum)){
                mapAB.set(sum,1)
            }else{
                mapAB.set(sum,mapAB.get(sum) + 1)
            }
        }
    }
    let res = 0
    for(let num3 of nums3){
        for(let num4 of nums4){
            let sum = 0  - (num4 + num3)
            if(mapAB.has(sum)){
                res += mapAB.get(sum)// res+= 1不正确,需要进行
            }
        }
    }
    return res
};

383. 赎金信

力扣题目链接(opens new window)

给定一个赎金信 (ransom) 字符串和一个杂志(magazine)字符串,判断第一个字符串 ransom 能不能由第二个字符串 magazines 里面的字符构成。如果可以构成,返回 true ;否则返回 false。

(题目说明:为了不暴露赎金信字迹,要从杂志上搜索各个需要的字母,组成单词来表达意思。杂志字符串中的每个字符只能在赎金信字符串中使用一次。)

注意:

你可以假设两个字符串均只含有小写字母。

canConstruct("a", "b") -> false
canConstruct("aa", "ab") -> false
canConstruct("aa", "aab") -> true

解题思路

  1. 暴力循环,太笨了不想写
  2. 使用哈希表 (可以使用map,也可以像《字母异位词》一样使用数组)

首先记录magazines 的字符以及其出现字数,存入map中,然后遍历ransomNote,对map中的值进行查找和更新,如果不存在需要的字母 或者 其次数小于0,说明不能构成,返回false.

var canConstruct = function(ransomNote, magazine) {
    let map = new Map()
    for(let char of magazine){
        if(map.has(char)){
            map.set(char,map.get(char)+1)
        }else{
            map.set(char,1)
        }
    }
    for(let char of ransomNote){
        if(map.has(char)){
            if(map.get(char) <=0){
                return false
            }
            map.set(char,map.get(char)-1)
        }else{
            return false
        }
    }
    return true
};

15. 三数之和

力扣题目链接(opens new window)

给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。

注意:  答案中不可以包含重复的三元组。

示例:

给定数组 nums = [-1, 0, 1, 2, -1, -4],

满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ]

解题思路

  1. 使用哈希表:比双指针遍历还要麻烦,蒜了
  2. 双指针法

具体可以查看 这里的动图

var threeSum = function(nums) {
    let res = []
    let left
    let right
    nums.sort((a,b) =>a-b)
    for(let i=0;i<nums.length;i++){
        if(nums[i] >0 ) break
        if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
        }
        left = i+1
        right = nums.length -1
        while(left < right){
            let sum = nums[i] + nums[left] + nums[right]
            if(sum > 0){
                right --
            }else if(sum< 0){
                left ++
            }else{
                res.push([nums[i],nums[left],nums[right]])
                while (right > left && nums[right] == nums[right - 1]){
                    right--
                } 
                while (right > left && nums[left] == nums[left + 1]) {
                    left++
                }
                left++
                right--
            }
        }
    }
    return res
};

18. 四数之和

力扣题目链接(opens new window)

题意:给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。

注意:

答案中不可以包含重复的四元组。

示例: 给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。 满足要求的四元组集合为: [ [-1, 0, 0, 1], [-2, -1, 1, 2], [-2, 0, 0, 2]

解题思路

在 三数之和 的基础上,在i 的前面多加一个 k, 多嵌套一层循环

注意剪枝操作: 1. 剪枝的时候,此时目标不为0,注意不能仅仅通过nums[i]>target就判断剪枝2. 注意去重的时候用continue,不是break

var fourSum = function(nums, target) {
    nums.sort((a,b) => a - b)

    let res = []
    
    for(let k=0;k<nums.length;k++){
        if(nums[k] > target && nums[k] >=0 ){
            break
        }

        if(k>0 && nums[k]===nums[k-1]){
            continue
        }

        for(let i=k+1;i<nums.length;i++){
            if(nums[i] + nums[k] > target && nums[i] + nums[k] >=0 ){
                break
            }
            if(i>k+1 && nums[i]===nums[i-1]){
                continue
            }
            let left = i+1
            let right = nums.length - 1

            while(left < right){
                if( nums[k]+nums[i]+nums[left] + nums[right] > target ){
                    right --
                }else if (nums[k]+nums[i]+nums[left] + nums[right] < target){
                    left ++
                }else{
                    res.push([nums[k],nums[i],nums[left] ,nums[right]])
                    while(left < right && nums[right]===nums[right-1]){
                        right --
                    }
                    while(right > left && nums[left] === nums[left +1]){
                        left ++
                    }
                    left ++
                    right --
                }
            }
        }
    }

    return res
};