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