LeetCode454.四数相加Ⅱ

最新推荐文章于 2024-06-19 08:00:00 发布

快乐早睡

最新推荐文章于 2024-06-19 08:00:00 发布

阅读量590

点赞数 11

分类专栏： LeetCode 文章标签：哈希算法算法开发语言 java c++ leetcode

本文链接：https://blog.csdn.net/weixin_40909461/article/details/138259665

版权

LeetCode 专栏收录该内容

132 篇文章 0 订阅

订阅专栏

在这里插入图片描述

LeetCode454.四数相加Ⅱ

题目：

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

$0 <= i, j, k, l < n$
$n u m s 1 [i] + n u m s 2 [j] + n u m s 3 [k] + n u m s 4 [l] == 0$

示例：

输入： $n u m s 1 = [1, 2], n u m s 2 = [- 2, - 1], n u m s 3 = [- 1, 2], n u m s 4 = [0, 2]$
输出： $2$
解释： 两个元组如下：
- $(0, 0, 0, 1) - > n u m s 1 [0] + n u m s 2 [0] + n u m s 3 [0] + n u m s 4 [1] = 1 + (- 2) + (- 1) + 2 = 0$
- $(1, 1, 0, 0) - > n u m s 1 [1] + n u m s 2 [1] + n u m s 3 [0] + n u m s 4 [0] = 2 + (- 1) + (- 1) + 0 = 0$

解题思路：

定义一个哈希表，然后遍历nums1和nums2，用哈希表记录a+b出现的次数，再遍历nums3和nums4，若有0-(c+d)出现，则加到结果中。

版本一：Java

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        int res = 0;
        Map<Integer, Integer> hash = new HashMap<Integer, Integer>(); 
        for(int a : nums1)
            for(int b : nums2){
                int sum = a + b;
                hash.put(sum, hash.getOrDefault(sum, 0) + 1);
            }
        for(int c : nums3)
            for(int d : nums4)
                res += hash.getOrDefault(0 - c - d, 0);
        return res;
    }
}

版本二：C++

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int, int> hash;
        for(int a : nums1)
            for(int b : nums2)
                hash[a + b] ++;
        int res = 0;
        for(int c : nums3)
            for(int d : nums4)
                if(hash.find(0 - (c + d)) != hash.end()) 
                	res += hash[0 - (c + d)];
        return res;
    }
};