题目概述
对应力扣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
示例:
输入:nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
输出:2
解释:
两个元组如下:
(0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
(1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0
示例:
输入:nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]
输出:1
解题思路
在不考虑暴力解法的同时,我们可以有这样的思路:将前两个数组所有加和的结果进行记录存储(这个过程用两个for循环嵌套即可),然后再将后两组的加和结果和前两组的进行对比(每计算一次就对比一次),看是否有相反数,将互为相反数的进行计数,最后返回结果即可。
这其中也存在这一个问题:用什么数据结构去存储前两个数组的所有加和结果?
- 用Set显然不行,如果有两种情况的和是一样的,只会显示一个,后面进行计数的时候会产生误差。
- 用列表可以,不过效率不高。因为到最后去计数的时候,你要一个个地去遍历符合条件的相反数有几个。直接将时间复杂度上升了一个层级。
- 用map是最好的。我们可以将和的结果作为key,而将出现次数作为关联的值,如此我们最后计数的时候只需要进行简单的累加就可以了,完全不需要再去进行遍历来计数。
代码实现
public int fourSumCount(int[] a, int[] b, int[] c, int[] d) {
Map<Integer,Integer> box = new HashMap<>();
int result = 0;
for (int num1 : a) {
for (int num2 : b) {
box.put(num1 + num2, box.getOrDefault(num1 + num2, 0) + 1);
}
}
for (int num3 : c) {
for (int num4 : d) {
if (box.containsKey(-(num3 + num4))) {
result += box.get(-(num3 + num4));
}
}
}
return result;
看起来有点臃肿,我们可以优化以下代码:
class Solution {
public int fourSumCount(int[] a, int[] b, int[] c, int[] d) {
Map<Integer,Integer> box = new HashMap<>();
int result = 0;
for (int num1 : a) for (int num2 : b) box.put(num1 + num2, box.getOrDefault(num1 + num2, 0) + 1);
for (int num3 : c) for (int num4 : d) if (box.containsKey(-(num3 + num4))) result += box.get(-(num3 + num4));
return result;
}
}
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
文章由极客之音整理,本文链接:https://www.bmabk.com/index.php/post/122262.html