难度中等782
给你四个整数数组 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 = 02. (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0
示例 2:
输入:nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]输出:1
提示:
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
暴力法:如果不限时间的话这个方法是对的,但是怎么可能不限制时间捏??
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
int cnt=0;
unordered_map<int,int>mp;
for(int i=0;i<nums1.size();i++){
for(int j=0;j<nums2.size();j++){
for(int k=0;k<nums3.size();k++){
for(int l=0;l<nums4.size();l++){
if(nums1[i]+nums2[j]+nums3[k]+nums4[l]==0)cnt++;
}
}
}
}
return cnt;
}
};
思路:四数相加这道题要把时间复杂度降到(我能想到的)最低的话,就要使用哈希表,要巧用哈希表就可以把暴力解法的O(n^4)降到O(n^2)
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int,int>mp;
for(int i=0;i<nums1.size();i++){
for(int j=0;j<nums2.size();j++){
int s = nums1[i]+nums2[j];
mp[s]++;
}
}
int cont = 0;
for(int i=0;i<nums3.size();i++){
for(int j=0;j<nums4.size();j++){
int k = 0 - (nums3[i]+nums4[j]);
if(mp.find(k)!=mp.end())cont+=mp[k];
}
}
return cont;
}
};
上一篇:环球快消息!去ོ村ོ游ོ吧ོ~
下一篇:最后一页
X 关闭
Copyright 2015-2022 华东经济日报网版权所有 备案号:京ICP备2022016840号-41 联系邮箱:2 913 236 @qq.com