HashMap使用-LeetCode做题总结 454. 四数相加 II

454. 四数相加 II

  • 最初思路
  • 优化思路
  • Java语法
    • 增强for的使用场景

最初思路

枚举,因为是要计算有多少个元组,所以每个元素肯定都要遍历到,所以干脆算出所有元组的和。
我想用四个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;
    }
}

HashMap使用-LeetCode做题总结 454. 四数相加 II_第1张图片

Java语法

增强for的使用场景

使用范围:用来遍历集合和数组(必须有遍历目标,目标只能是集合或者数组),所有单列表集合都可以使用增强for循环。

你可能感兴趣的:(LeetCode做题总结,leetcode)