您当前的位置:首页 >> 要闻 > > 
今日快看!力扣:454. 四数相加 II
来源: 哔哩哔哩      时间:2023-04-09 15:51:51

454. 四数相加 II

难度中等782

给你四个整数数组 nums1nums2nums3和 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 关闭

X 关闭

观点