力扣:454. 四數(shù)相加 II
454. 四數(shù)相加 II
難度中等782
給你四個(gè)整數(shù)數(shù)組?nums1
、nums2
、nums3
?和?nums4
?,數(shù)組長(zhǎng)度都是?n
?,請(qǐng)你計(jì)算有多少個(gè)元組?(i, j, k, l)
?能滿(mǎn)足:
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解釋?zhuān)簝蓚€(gè)元組如下: 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:
輸入: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
第一種錯(cuò)法:
暴力法:如果不限時(shí)間的話(huà)這個(gè)方法是對(duì)的,但是怎么可能不限制時(shí)間捏??
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;
????}
};
第一種對(duì)法:
思路:四數(shù)相加這道題要把時(shí)間復(fù)雜度降到(我能想到的)最低的話(huà),就要使用哈希表,要巧用哈希表就可以把暴力解法的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;
????}
};