代码随想录算法训练营第六天| 454.四数相加II、383. 赎金信、15. 三数之和、18. 四数之和
454.四数相加II
【题目描述】
给你四个整数数组 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.首先定义 一个unordered_map,key放a和b两数之和,value 放a和b两数之和出现的次数。
2.遍历大A和大B数组,统计两个数组元素之和,和出现的次数,放到map中。
3.定义int变量count,用来统计 a+b+c+d = 0 出现的次数。
4.在遍历大C和大D数组,找到如果 0-(c+d) 在map中出现过的话,就用count把map中key对应的value也就是出现次数统计出来。
5.最后返回统计值 count 就可以了
哈希表法:
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int,int> map; //key保存nums1加nums2的值,value保存key出现的次数。
for(auto a : nums1){
for(auto b : nums2){
++map[a + b];
}
}
int count = 0; //count记录四数之和等于0出现的次数。
for(auto c : nums3){
for(auto d : nums4){
auto iter = map.find(0 - (c + d));
if(iter != map.end()){
count += map[0 - (c + d)];
}
}
}
return count;
}
};
时间复杂度:O(n^2)
空间复杂度:O(n^2),最坏情况下A和B的值各不相同,相加产生的数字个数为 n^2
383. 赎金信
【题目描述】
给你两个字符串:ransomNote 和 magazine ,判断 ransomNote 能不能由 magazine 里面的字符构成。
如果可以,返回 true ;否则返回 false 。
magazine 中的每个字符只能在 ransomNote 中使用一次。
【思路】
1.首先1 <= ransomNote.length, magazine.length <= 10^5,考虑到可以使用数组vInt,保存ransomNote中字符出现的次数。
2.遍历字符串magazine中的字符,将字符串magazine中出现的字符在vInt中的次数减一。
3.最后如果数组vInt中没有大于0的数,表示可以,返回true,否则返回false。
哈希表法:
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
vector<int> vInt(26,0);
for(auto i : ransomNote){
++vInt[i - 'a']; //保存字符串ransomNote中字符出现的次数
}
for(auto i : magazine){
--vInt[i - 'a']; //减去字符串magazine中字符出现的次数
}
for(int i = 0;i != vInt.size();++i){
if(vInt[i] > 0){ //如果数组vInt有大于0的数,表示不可以,返回false
return false;
}
}
//否则表示可以,返回true
return true;
}
};
时间复杂度:O(n+m) ,n是ransomNote长度,m是magazine长度
空间复杂度:O(1)
15. 三数之和
【题目描述】
给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请
你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
【思路】
1.使用双指针法
双指针法:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> result;
sort(nums.begin(),nums.end());
for(int i = 0;i != nums.size();++i){
//如果num[i]大于0,则不再存在三元组
if(nums[i] > 0){
break;
}
//对nums[i]去重
if(i > 0 && nums[i] == nums[i-1]){
continue;
}
int left = i + 1;
int right = nums.size() - 1;
while(left < right){
if(nums[i] + nums [right] + nums[left] > 0){
--right;
}
else if(nums[i] + nums[right] +nums[left] < 0){
++left;
}
else{
result.push_back({nums[i],nums[left],nums[right]});
//找到三元组后,对nums[j]和nums[k]去重
while(left < right && nums[left] == nums[left + 1]){
++left;
}
while(left < right && nums[right] == nums[right - 1]){
--right;
}
++left;
--right;
}
}
}
return result;
}
};
时间复杂度:O(n^2)
空间复杂度:O(1)
18. 四数之和
【题目描述】
给你一个由 n 个整数组成的数组 nums ,和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] (若两个四元组元素一一对应,则认为两个四元组重复):
0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按 任意顺序 返回答案 。
【思路】
1.双指针法。和三数之和一样。但是多一层循环,还有防止四数之和溢出要是用long类型转换。
双指针法:
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> result;
sort(nums.begin(),nums.end());
for(int k = 0;k != nums.size(); ++k){
if(nums[k] > target && nums[k] >= 0){
break;
}
if(k > 0 && nums[k] == nums[k - 1]){
continue;
}
for(int i = k + 1;i != nums.size();++i){
if(nums[i] + nums[k] > target && (nums[i] + nums[k]) >= 0){
break;
}
if(i > k + 1 && nums[i] == nums[i - 1]){
continue;
}
int left = i + 1;
int right = nums.size() - 1;
while(left < right){
//nums[k] + nums[i] + nums[left] + nums[right]会溢出
if((long)nums[k] + nums[i] + nums[left] + nums[right] < target){
++left;
}
else if((long)nums[k] + nums[i] + nums[left] + nums[right] > target){
--right;
}
else{
result.push_back({nums[k],nums[i],nums[left],nums[right]});
while(left < right && nums[left] == nums[left + 1]){
++left;
}
while(left < right && nums[right - 1] == nums[right]){
--right;
}
++left;
--right;
}
}
}
}
return result;
}
};
时间复杂度:O(n^3)
空间复杂度:O(1)