力扣 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示例 1：
输入：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
3. 
示例 2：
输入：nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]
输出：1

思想

遍历 nums1 和 nums2：
双重循环遍历 nums1 和 nums2 的每个元素 i 和 j。计算和 i+j，并将该和作为键存储到 map 中，值为这个和出现的次数（如果已存在，则增加计数）。
遍历 nums3 和 nums4：
再次使用双重循环，这次遍历 nums3 和 nums4 的每个元素 k 和 q。
计算 - (k + q)，即我们需要在 map 中找到的目标值，使得四个数之和为零。
使用 map.find(target) 查找是否有符合条件的键值对存在。如果找到了，将找到的值（即前面计算出的和的出现次数）加到 count 中。

举例说明：

第一步：计算 nums1 和 nums2 的所有可能的元素和
循环过程：
对于 nums1[0] = 1：nums2[0] = -2 -> 和为 -1，map[-1] = 1nums2[1] = -1 -> 和为 0，map[0] = 1
对于 nums1[1] = 2：nums2[0] = -2 -> 和为 0，map[0] 现在增加为 2（因为已经有一个和为 0）nums2[1] = -1 -> 和为 1，map[1] = 1
此时，map 的内容如下：-1 出现 1 次0 出现 2 次1 出现 1 次第二步：计算 nums3 和 nums4 的所有可能的元素和的相反数，并查找是否存在于 map 中
循环过程：
对于 nums3[0] = -1：nums4[0] = 0 -> 和为 -1，相反数是 1，map 中 1 出现了 1 次，所以 count += 1nums4[1] = 2 -> 和为 1，相反数是 -1，map 中 -1 出现了 1 次，所以 count += 1
对于 nums3[1] = 2：nums4[0] = 0 -> 和为 2，相反数是 -2，map 中不存在 -2nums4[1] = 2 -> 和为 4，相反数是 -4，map 中不存在 -4
结果：count 的值为 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

完整代码

#include<iostream>
#include<vector>
#include<unordered_map>
using namespace std;class Solution {
public:int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {unordered_map<int, int> map;int count = 0;for(int i : nums1){for(int j : nums2){map[i + j]++;}}for(int k : nums3){for(int q : nums4){int target = 0 - (k + q);if(map.find(target) != map.end()){count += map[target];}}}return count;}
};int main()
{Solution s;vector<int> nums1 = {1,2};vector<int> nums2 = {-2,-1};vector<int> nums3 = {-1,2};vector<int> nums4 = {0,2};cout << s.fourSumCount(nums1, nums2, nums3, nums4) << endl;return 0;
}