HashMap使用-LeetCode做题总结 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循环。