算法合集 | 哈希表 | Leetcode 454.四数相加II

108 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第12天,点击查看活动详情

前言

本系列文章主要会总结一些常见的算法题目以及算法的易错点,难点,以及一些万用的公式,并且结合实际的 Leetcode 题目来进行加深理解以及实际应用,算法这种东西,属于是一到用时方恨少的类型,在平时总结一些常见的简单算法,经常磨练自己的算法思维,对于日常的开发还是能有不少的帮助的。

Leetcode 454.四数相加II

给你四个整数数组 nums1、nums2、nums3 和 nums4 ,数组长度都是 n ,请你计算有多少个元组 (i, j, k, l) 能满足:

  • 0 <= i, j, k, l < n
  • nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

经典的哈希表应用题,因为数据分别来自于4个数组,所以不需要考虑是否会重复的问题,方法有点类似于 1. 两数之和 的做法,4个数组也是能够通过4个循环嵌套解决的,利用哈希表就是对这一个过程进行优化

首先来看一下暴力的方法

image.png

直接超出时间限制了,那么就像之前说的,一旦需要快速的找到某个对象中的某个元素,就可以想到使用哈希表,利用哈希表来进行优化,把四个数组分为两个两个,我们先把前两个的所有组合情况都通过一张哈希表保存起来,然后在需要使用的时候,就可以从表中快速的取到,拿去和后两个数组的所有可能情况作比较,这样子的话,时间复杂度就会由 O(n4) 降到 O(n2) 。

function fourSumCount(nums1: number[], nums2: number[], nums3: number[], nums4: number[]): number {
    let res = 0
    let map = new Map()
    nums1.forEach(item1 => {
        nums2.forEach(item2 => {
            map.set(item1+item2,(map.get(item1+item2) || 0) + 1)
        })
    })
    nums3.forEach(item3 => {
        nums4.forEach(item4 => {
           if(map.get(0-item3-item4)) res+=map.get(0-item3-item4)
        })
    })
    return res
};

图片.png