「代码随想录算法训练营」第六天 | 哈希表 part2
454. 四数相加 II
题目链接:https://leetcode.cn/problems/4sum-ii/
题目难度:中等
文章讲解:https://programmercarl.com/0454.四数相加II.html
视频讲解: https://www.bilibili.com/video/BV1Md4y1Q7Yh
题目状态:没思路
思路:
- 创建一个
unordered_map<int, int>
类型的哈希表用来存储前两个数组中的元素相加之和,键为和,值为次数; - 创建一个
int
类型的变量count
,用来存储四个数组中元素之和等于0
的次数; - 遍历后两个数组之和,若哈希表中存在一个元素和这两个数组中元素之和相加为
0
的元素,将其次数加入count
; - 返回
count
。
实现代码:
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int, int> umap;
for(auto &num1 : nums1) {
for(auto &num2 : nums2) {
umap[num1 + num2]++;
}
}
int count = 0;
for(auto &num3 : nums3) {
for(auto &num4 : nums4) {
if(umap.find(0 - (num3 + num4)) != umap.end()) {
count += umap[0 - (num3 + num4)];
}
}
}
return count;
}
};
383. 赎金信
题目链接:https://leetcode.cn/problems/ransom-note/
题目难度:简单
文章讲解:https://programmercarl.com/0383.赎金信.html
题目状态:一遍过
思路:
- 若
magazine
的长度小于ransomNote
的长度,则必然不符合要求,直接返回false
; - 创建一个
unordered_map<int, int>
类型的map
,用来存储magazine
中每个字母出现的次数; - 之后遍历
ransomNote
,使用map
存储ransomNote
中出现的字母,这时每出现一次,map
中对应的键值对的值需要-1
; - 判断
map
中是否有负数,有负数代表这不符合情况,返回false
; - 没有负数就返回
true
。
代码实现:
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
int ransomLen = ransomNote.size();
int magLen = magazine.size();
if(ransomLen > magLen) return false;
unordered_map<int, int> umap(26);
for(auto &mag : magazine) {
umap[mag - 'a']++;
}
for(auto &ransom : ransomNote) {
umap[ransom - 'a']--;
}
for(int i = 0; i < 26; ++i) {
if(umap[i] < 0) return false;
}
return true;
}
};
15. 三数之和
题目链接:https://leetcode.cn/problems/3sum/
题目难度:中等
文章讲解:https://programmercarl.com/0015.三数之和.html
视频讲解: https://www.bilibili.com/video/BV1GW4y127qo
题目状态:没思路,啊啊啊!梦又破碎的地方!
思路:
- 首先对数组进行排序;
- 定义一个
vector<vector<int>>
类型的二维数组res
用来存放答案; - 下面直接看动图更有思路,主要看代码细节。
代码实现:
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(nums[i] > 0) {
return res;
}
if(i > 0 && nums[i] == nums[i - 1]) {
continue;
}
int left = i + 1;
int right = nums.size() - 1;
while(right > left) {
if(nums[i] + nums[right] + nums[left] > 0) right--;
else if(nums[i] + nums[right] + nums[left] < 0) left++;
else {
res.push_back(vector<int>{nums[i], nums[left], nums[right]});
while(right > left && nums[right] == nums[right - 1]) right--;
while(right > left && nums[left] == nums[left + 1]) left++;
right--;
left++;
}
}
}
return res;
}
};
18. 四数之和
题目链接:https://leetcode.cn/problems/4sum/
题目难度:中等
文章讲解:https://programmercarl.com/0018.四数之和.html
视频讲解: https://www.bilibili.com/video/BV1DS4y147US
题目状态:按照前面一题的思路,倒在了long
类型转换上
思路:
在上面一题的基础上加一层for
循环
实现代码:
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> res;
sort(nums.begin(), nums.end());
for(int i = 0; i < nums.size(); ++i) {
if(nums[i] > target && nums[i] >= 0) break;
if(i > 0 && nums[i] == nums[i - 1]) continue;
for(int j = i + 1; j < nums.size(); ++j) {
if((nums[i] + nums[j]) > target && (nums[i] + nums[j]) >= 0) break;
if(j > i + 1 && nums[j] == nums[j - 1]) continue;
int left = j + 1;
int right = nums.size() - 1;
while(right > left) {
if((long) nums[i] + nums[j] + nums[left] + nums[right] > target) right--;
else if((long) nums[i] + nums[j] + nums[left] + nums[right] < target) left++;
else {
res.push_back(vector<int>{nums[i], nums[j], nums[left], nums[right]});
while(right > left && nums[right] == nums[right - 1]) right--;
while(right > left && nums[left] == nums[left + 1]) left++;
right--;
left++;
}
}
}
}
return res;
}
};