0
点赞
收藏
分享

微信扫一扫

数据结构与算法【LeetCode-Offer】:3.6哈希表—四数相加Ⅱ

火热如冰 2022-02-08 阅读 54

四数相加 II

📃 题目描述

题目链接:

在这里插入图片描述

🔔 解题思路

注意返回值是满足条件的元组的个数。

本题涉及四个数组,暴力解法就是四层循环,显然不可取,我们考虑两两处理的方式,具体步骤如下:

首先定义变量 count,用来统计符合条件的元组的个数,即(a+b)+(c+d)= 0 出现的次数,然后:

1)先处理 A 数组和 B 数组

  • 定义一个 Map,key 存放 a 和 b 两数之和,value 存放 a 和 b 两数之和出现的次数
  • 遍历 A 和 B 数组,统计两个数组元素之和以及出现的次数,放到 map 中

2)再处理 C 数组和 D 数组

  • 遍历 C 和 D 数组,如果 0-(c+d) 这个 key 在 map 中出现过的话,即表示这 a b c d 是一个符合条件的元组,count 加上 map 中 key 是 0-(c+d) 对应的 value 即可
package com.kami.leetcode.hash_table;

import java.util.HashMap;

/**
 * @Description: TODO
 * @author: scott
 * @date: 2022年02月08日 9:45
 */
public class Solution_454 {

    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4){

        int count = 0;

        // key 存放 a 和 b 两数之和,value 存放 a 和 b 两数之和出现的次数
        HashMap<Integer, Integer> map = new HashMap<>();


        //处理A数组和B数组
        for (int a : nums1) {
            for (int b : nums2) {
                int temp = a + b;
                map.put(temp, map.getOrDefault(temp, 0) + 1);
            }
        }

        // 再处理 C 数组和 D 数组
        for (int c : nums3) {
            for (int d : nums4) {
                int temp = c + d;
                // 如果 0-(c+d) 这个 key 在 map 中出现过的话, 即表示这 a b c d 是一个符合条件的元组
                if(map.containsKey(0 - temp)){
                    count += map.get(0 - temp);
                }
            }
        }

        return count;
    }
}

举报

相关推荐

0 条评论