• LeetCode 454. 4Sum II


    LeetCode 454. 4Sum II (四数相加 II)

    题目

    链接

    https://leetcode.cn/problems/4sum-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

    示例

    输入:nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
    输出:2
    解释:
    两个元组如下:

    1. (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
    2. (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0
      示例 2:

    提示

    n == nums1.length
    n == nums2.length
    n == nums3.length
    n == nums4.length
    1 <= n <= 200
    -228 <= nums1[i], nums2[i], nums3[i], nums4[i] <= 228

    思路

    哈希表来存储12中的和,然后34中的和用来查找哈希表,这样可以确保n2,对于这类问题,可以采用一半一半的方法,如果是6个数组,那就可以用三三分组,n3的复杂度。

    复杂度分析

    时间复杂度 O(n2)
    空间复杂度 O(n)
    

    代码

    Java

        public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
            int n = nums1.length;
            int ans = 0;
            HashMap<Integer, Integer> map = new HashMap<>();
            for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                    int s = nums1[i] + nums2[j];
                    map.put(s, map.getOrDefault(s, 0) + 1);
                }
            }
            for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                    int s = nums3[i] + nums4[j];
                    ans += map.getOrDefault(-s, 0);
                }
            }
            return ans;
        }
    
  • 相关阅读:
    爬虫
    Django
    python多线程
    python基础
    深度学习_1_Tensorflow_1
    人工智能_5_决策树_随机森林
    人工智能_4_k近邻_贝叶斯_模型评估
    人工智能_3_机器学习_概述
    Python re 模块
    Python函数式编程
  • 原文地址:https://www.cnblogs.com/blogxjc/p/16288060.html
Copyright © 2020-2023  润新知