LeetCode 第454题 四数相加2 做题记录

题目描述

我的解法

思路

最直接的暴力求解有四层嵌套循环，遂不考虑；
而联想到之前做的第1题两数之和中采用的HashMap结构
而1中使用HashMap结构相当于只能包含两个数组的信息，所以就想到了先对A和B的和构建哈希表（相当于把A和B作为一个整体），Key值为两数组元素的和，Value为和的值出现的次数，然后与C和D的和进行比较（把C和D作为整体）

对应Java代码

错误提示：因为像在python中用数组取值一样的方法在HashMap中取值，而应该用.get(key）这个方法

class Solution {
    public int fourSumCount(int[] A, int[] B, int[] C, int[] D) {
        Map<Integer, Integer> addAB = new HashMap<>();
        for(int i = 0; i < A.length; i++){
            for(int j = 0; j < B.length; j++){
                if(addAB.containsKey(A[i] + B[j])){ 
                    int tmp = addAB.get(A[i] + B[j]) + 1;
                    addAB.put((A[i] + B[j]), tmp);
                    }
                else{ addAB.put((A[i] + B[j]), 1); }
            }
        }
        int result = 0;
        for(int i = 0; i < C.length; i++){
            for(int j = 0; j < D.length; j++){
                if(addAB.containsKey(0 - C[i] - D[j])) { 
                    result += addAB.get(0 - C[i] - D[j]); 
                    }
            }
        }
        return result;
    }
}

复杂度分析

时间复杂度：O(n²) 我们使用了两次二重循环，时间复杂度均为 O(n²) 。在循环中对哈希映射进行的修改以及查询操作的期望时间复杂度均为 O(1) ，因此总时间复杂度为 O(n²)。

空间复杂度：O(n²)，即为哈希映射需要使用的空间。在最坏的情况下，
A[i]+B[j] 的值均不相同，因此值的个数为 n²，也就需要O(n²)的空间

收获总结

在构建HashMap时要根据题目考虑结构中存储的Key和对应的value分别是什么