DAY 37 leetcode 454--哈希表.四数相加

Fantasydg2025-04-07 0:02

题号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

解法

java 复制代码

class Solution {
    public int fourSumCount(int[] A, int[] B, int[] C, int[] D) {
        Map<Integer,Integer>map=new HashMap<>();
        int n=A.length;
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
               map.put(A[i]+B[j],map.getOrDefault(A[i]+B[j],0)+1);
            }
        }
        int count=0;
        for(int k=0;k<n;k++){
            for(int l=0;l<n;l++){
                if(map.containsKey(-(C[k]+D[l])))
                count+=map.get(-(C[k]+D[l]));
            }
        }
        return count;
    }
}

思路

将AB CD分为两组，先遍历AB 将两者之和作为Key，出现的次数作为value存进HashMap中，然后再遍历CD，若出现-Key，则说明可以与AB中的元素相加为0，此时count+=value，即可记录所有相加为0的次数了