四数相加 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;
    }
}