HashMap使用-LeetCode做题总结 454. 四数相加 II
454. 四数相加 II
最初思路
枚举,因为是要计算有多少个元组,所以每个元素肯定都要遍历到,所以干脆算出所有元组的和。
我想用四个for循环加,但是失败。
优化思路
参考力扣
四数相加为零,a+b+c+d=0,那么 a+b = -(c+d),这个等式可以利用哈希表。
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
Map<Integer,Integer> countAB = new HashMap<Integer,Integer>();
for(int v : nums1) {
for(int u : nums2) {
countAB.put(u+v,countAB.getOrDefault(u+v,0)+1);
}
}
int ans=0;
for(int v : nums3) {
for(int u : nums4) {
if(countAB.containsKey(-(v+u))) {
ans += countAB.get(-(v+u));
}
//countAB.put(u+v,countAB.getOrDefault(u+v,0)+1);
}
}
return ans;
}
}
Java语法
增强for的使用场景
使用范围:用来遍历集合和数组(必须有遍历目标,目标只能是集合或者数组),所有单列表集合都可以使用增强for循环。