一、今日刷题
1. 第六部分:哈希表 – 454.四数相加 II
答案代码:
分组 + 哈希表
package HashSet;
import java.util.HashMap;
import java.util.Map;
/**
* @author: LYZ
* @date: 2022/2/13 16:36
* @description: 454.给你四个整数数组 nums1、nums2、nums3 和 nums4 ,数组长度都是 n ,请你计算有多少个元组 (i, j, k, l) 能满足:
* 0 <= i, j, k, l < n
* nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
*/
public class FourSumCount {
public static void main(String[] args) {
int[] nums1 = {1, 2};
int[] nums2 = {-2, -1};
int[] nums3 = {-1, 2};
int[] nums4 = {0, 2};
FourSumCount fourSumCount = new FourSumCount();
int ans = fourSumCount.fourSumCount(nums1, nums2, nums3, nums4);
System.out.println(ans);
}
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
Map<Integer, Integer> map = new HashMap<>();
int count = 0;
for (int num1 : nums1) {
for (int num2 : nums2) {
int addNumber = num1 + num2;
int frequency = map.getOrDefault(addNumber, 0);
map.put(addNumber, frequency + 1);
}
}
for (int num3 : nums3) {
for (int num4 : nums4) {
int checkNumber = -(num3 + num4);
if (map.containsKey(checkNumber)) {
int addCount = map.get(checkNumber);
count += addCount;
}
}
}
return count;
}
}