Leetcode P454 Java使用Hash表解决
ideas
由题意可知nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
,我们可以分解为(nums1[i] + nums2[j]) + (nums3[k] + nums4[l]) == 0
我们使用HashMap来记录nums1[i] + nums2[j]相加的结果,每个结果个出现了多少次
for (int i : nums1) {
for (int j : nums2) {
long tmp = i + j;
map.put(tmp, map.getOrDefault(tmp, Long.valueOf(0))+1);
}
}
接下来判断0 - nums3[k] + nums4[l]是否在map中存在,如果存在则res+= map.get(0 - nums3[k] + nums4[l] )出现的个数,
关于为什么是0 - nums3[k] + nums4[l]?
因为,假设nums1[i] + nums2[j]等于12,那么判断nums3[k] + nums4[l] 的值等于-12,0- (-12)查看12是否在map中存在,如果存在则res+= map.get(0 - nums3[k] + nums4[l] )出现的个数,
for (int i : nums3) {
for (int j : nums4) {
long tmp = i+j;
if (map.containsKey(0 - tmp)) {
res+=map.get(0-tmp);
}
}
}
code
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
int res =0;
HashMap<Long,Long> map = new HashMap<>();
for (int i : nums1) {
for (int j : nums2) {
long tmp = i + j;
map.put(tmp, map.getOrDefault(tmp, Long.valueOf(0))+1);
}
}
for (int i : nums3) {
for (int j : nums4) {
long tmp = i+j;
if (map.containsKey(0 - tmp)) {
res+=map.get(0-tmp);
}
}
}
return res;
}
}