题目描述
简单来说就是给四个数组,每个数组拿一个数,看看有多少组合(元素相同数组不同也算不同)
分析
个人心路历程
这个题我一拿到无从下手,当然最暴力的方法四个for循环肯定能做出来,但是复杂度太高了。就想怎么把复杂度降低,在看carl哥视频的时候听到两个做组合放map里面瞬间恍然大悟,这样时间复杂度就会变为平方。然后在map里面进行搜索
代码实现
注意!!!大家把数组的编号看对,我因为这个报错四五次
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
int count = 0; //定义计数器
map<int, int> add1; // 定义map
// 前两个相加放在map里面
for(int i=0; i<nums1.size(); i++)
{
for(int j=0; j<nums2.size(); j++)
{
add1[nums1[i]+nums2[j]]++;
}
}
// 后面两个相加再上面的里面进行搜索需要的数
for(int i=0; i<nums3.size(); i++)
{
for(int j=0; j<nums4.size(); j++)
{
if(add1.find(0-(nums3[i]+nums4[j])) != add1.end())
{
// 找到则count加add1中的value值
count = count + add1[0-(nums3[i]+nums4[j])];
}
}
}
return count;
}
};
细节
第一次我没怎么想,map定义用的是底层红黑树实现的 key有序 所以效率没unordered_map的效率高
leetcode运行如图
当我修改之后
两次对比有了明显改善