0哈希中等 LeetCode454. 四数相加 II

最新推荐文章于 2024-07-20 22:34:59 发布

18阿鲁

最新推荐文章于 2024-07-20 22:34:59 发布

阅读量97

点赞数

文章标签：哈希算法 leetcode 散列表

本文链接：https://blog.csdn.net/weixin_43260719/article/details/125353228

版权

454. 四数相加 II

描述

给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：

0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

分析

暴力解法的时间复杂度是O（n^4）
把四层循环拆分成2个O(n^2)的循环

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        Map<Integer,Integer> map = new HashMap<>();
        for (int i = 0; i < nums1.length; i++) {
            for (int j = 0; j < nums2.length; j++) {
                int sum = nums1[i]+nums2[j];
                map.put(sum,map.getOrDefault(sum,0)+1);
            }
        }
        int ans = 0;
        for (int i = 0; i < nums3.length; i++) {
            for (int j = 0; j < nums4.length; j++) {
                int sum = nums3[i]+nums4[j];
                int tmp = map.getOrDefault(-sum,0);
                ans += tmp;
            }
        }
        return ans;
    }
}