代码随想录算法训练营第七天| 454.四数相加II、383. 赎金信、15. 三数之和、18. 四数之和
454.四数相加II
老样子,先看代码:
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int,int> cnt;
for(auto n3:nums3)
for(auto n4:nums4)
cnt[n3+n4]++;
int res = 0;
for(auto n1:nums1)
for(auto n2:nums2)
res += cnt[-(n1+n2)];
return res;
}
};
这道题目和前面letcoded第一题两数之和差不多,但是也是相对麻烦一点,就是这里是四个数组,首先说写一个四重循环行不行,n的范围是1~200,四重循环就是1.6 * 109 这样会超时,所以最好降到n2 的时间复杂度,所以我们可以两个数组,两个数组的看,首先看nums3和nums4,先计算这两个数组中所有数字组合相加的和,然后用一个哈希表存储每个和出现的次数,对于一个n1+n2,有多少个-(n3+n4)在,就有多少种可能,这样就计算出满足条件的元组个数了。
383. 赎金信
看代码:
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
unordered_map<char,int> cnt;
for(auto m:magazine)
cnt[m]++;
for(auto r:ransomNote)
{
cnt[r]--;
if(cnt[r] < 0)
return false;
}
return true;
}
};
这道题目说看看magazine中的字符能否组成ransomNote,所以只要保证magazine中有ransomNote的出现的所有字母,而且出现的次数大于等于ransomNote中就可以。所以我们先对magazine中字母进行统计,然后对ransomNote进行检查,相当于ransomNote在使用magazine的字符,magazine中的字符用一个少一个,所以就进行"–"操作,如果次数小于0说明magazine中该字符数量不够或者根本就没有。那就返回false。
15. 三数之和
先看代码:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
sort(nums.begin(),nums.end());
for(int i = 0;i < nums.size();i++)
{
if(i && nums[i] == nums[i - 1]) continue;
for(int j = i + 1,k = nums.size() - 1;j < k;j++)
{
if(j > i+1 && nums[j] == nums[j - 1]) continue; //j不能和i重叠
while(j < k - 1 && nums[i] + nums[j] + nums[k - 1] >= 0) k--; //如果k的下一个数和j没有重叠
if(nums[i] + nums[j] + nums[k] == 0)
{
res.push_back({nums[i],nums[j],nums[k]});
}
}
}
return res;
}
};
这道题目用双指针好一些,我用哈希表试着做了,有点麻烦,所以这里就说双指针吧,数组用双指针,需要数组是有序的,所以首先上来排序,然后是思路,首先我先定下i,然后对j和k进行移动,i < j < k,三者不能有重叠,j从左往右走,k从右往左走,只要相遇就停下,这里我们可以观察这个式子:nums[i] + nums[j] + nums[k],现在数组是有序的,nums[i]是固定的,所以当j向右走的时候,nums[j]就在增大,同时为了保证nums[i] + nums[j] + nums[k] = 0,nums[j]在增大,也就意味着nums[k]就要减小,所k从右往左走,当我nums[i] 和nums[j]固定了的时候(也就是while()循环里的式子),nums[k]是在减小的,每次我们尝试减小nums[k],只要nums[i] + nums[j] + nums[k - 1],也就是k的下一个数仍然满足条件,就继续,直到ums[i] + nums[j] + nums[k - 1] < 0,这个时候nums[i] + nums[j] + nums[k]不是0,那也是距离0最近的一个正数了,我们只要等于0的即可。
说不好,建议看看y总视频:
https://www.acwing.com/activity
18. 四数之和
看代码:
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
sort(nums.begin(),nums.end());
vector<vector<int>> res;
for(int i = 0;i < nums.size();i++)
{
if(i && nums[i] == nums[i - 1]) continue;
for(int j = i + 1;j < nums.size();j++)
{
if(j > i+1 && nums[j] == nums[j - 1]) continue;
for(int k = j + 1,u = nums.size() - 1;k < u;k++)
{
if(k > j+1 && nums[k] == nums[k - 1]) continue;
while(k < u - 1 && (long)nums[i] + nums[j] + nums[k] + nums[u - 1] >= target) u--;
if(((long)nums[i] + nums[j] + nums[k] + nums[u]) == target)
res.push_back({nums[i],nums[j],nums[k],nums[u]});
}
}
}
return res;
}
};
原理和三个数之和一样,这次是i,j,k在左边,u从右往左走,不过这个题的测试用例是真坑啊,真坑啊,真坑啊,注意中间的long类型。