代码随想录算法训练营第7天 | 454.四数相加Ⅱ、383.赎金信、15.三数之和、18.四数之和
文章目录
454.四数相加Ⅱ
解题思路
对四个数组进行两两拆分,先统计 num1 和 num2 的各个元素之和和它们出现的次数,再遍历 num3 和 num4,就可以将本题转化为两数之和。
Step1:定义 unordered_map,key 用来存放前两个数组元素和,value 存放元素和出现的次数。
Step2:遍历前两个数组,将和出现的次数存放到 map 中。
Step3:遍历后两个数组,将满足条件的情况记录到 count 中 。
代码实现
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3,
vector<int>& nums4) {
unordered_map<int, int> map;
for (int a : nums1) {
for (int b : nums2) {
map[a + b]++;
}
}
int count = 0;
for (int c : nums3) {
for (int d : nums4) {
if (map.find(0 - (c + d)) != map.end()) {
count += map[0 - (c + d)];
}
}
}
return count;
}
};
题目总结
本题不需要考虑有重复的四个元素相加等于0的情况,只需要找到四个数等于0即可。
383.赎金信
解题思路
本题和之前做过的有效的字母异位词很相似,字母异位词是要求两个字符串能否相互组成,而本题要求其中字符串中的字符能够组成另一个字符串,不要求相互组成。主要思路是先遍历 magazine,记录其中每个字母出现的次数,接着遍历 ransomNote,每遇到一个字母,在哈希中做减一操作,若数字被减少到0,则返回false,否则返回true。
代码实现
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
if (ransomNote.size() > magazine.size()) {
return false;
}
int record[26] = {0};
for (int i = 0; i < magazine.size(); i++) {
record[magazine[i] - 'a']++;
}
for (int i = 0; i < ransomNote.size(); i++) {
record[ransomNote[i] - 'a']--;
if (record[ransomNote[i] - 'a'] < 0) {
return false;
}
}
return true;
}
};
题目总结
本题也可以使用 map,但是 map 空间消耗比较大,数组会更加的简单高效。
15.三数之和
解题思路
- 哈希解法
与两数之和思路相似,确定 0-(a+b) 是否在数组中出现过,将符合条件的三元组放到 vector 中,题目中要求三元组不可重复马,所以需要再次进行去重操作。
- 双指针解法
首先对数组中元素进行排序处理。遍历数组,将一个指针 left 放在 i+1 位置,另一个 right 在末尾。进行 while 循环,条件为 left<right,若三数之和大于0,指针向左移;和小于0,指针向右移;等于0,记录值。两个指针同时移动,进行去重处理。
代码实现
双指针解法
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++) {
if (nums[i] > 0) {
return result;
}
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[left] + nums[right] > 0) {
right--;
} else if (nums[i] + nums[left] + nums[right] < 0) {
left++;
} else {
result.push_back(
vector<int>{nums[i], nums[left], nums[right]});
while (left < right && nums[left] == nums[left + 1]) {
left++;
}
while (left < right && nums[right] == nums[right - 1]) {
right--;
}
left++;
right--;
}
}
}
return result;
}
};
题目总结
注意几个剪枝操作:在 for 循环遍历时,若前面已经遍历的元素就跳过;排序之后若数组第一个元素已经大于0,那么可以直接返回结果,因为无论如何也不可能组成三元组。
18.四数之和
解题思路
和三数之和思路完全一样,使用双指针法进行解题,两层 for 循环 nums[i]+nums[k] 为确定值。注意力扣测试用例中存在高位数,所以需要找到 (long)nums[i]+nums[k]+nums[left]+nums[right]==target 的情况。
代码实现
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> result;
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size(); i++) {
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
for (int k = i + 1; k < nums.size(); k++) {
if (k > i + 1 && nums[k] == nums[k - 1]) {
continue;
}
int left = k + 1;
int right = nums.size() - 1;
while (left < right) {
if ((long)nums[i] + nums[k] + nums[left] + nums[right] >
target) {
right--;
} else if ((long)nums[i] + nums[k] + nums[left] +
nums[right] <
target) {
left++;
} else {
result.push_back(
{nums[i], nums[k], nums[left], nums[right]});
while (left < right && nums[right] == nums[right - 1]) {
right--;
}
while (left < right && nums[left] == nums[left + 1]) {
left++;
}
right--;
left++;
}
}
}
}
return result;
}
};
题目总结
三数之和和四数之和的关键都在于剪枝和去重的操作。