454.四数相加II
题目链接: 454.四数相加II
思路
从每个数组中找出一个数相加和为0,找找满足的元组数,只是要求坐标不能重复,数值可以重复。
- 此题首先可以想到用暴力解法,循环遍历每个数组,找出满足条件的元组,但是时间复杂度为O(n^4);
- 可以考虑将四个数组分组,然后再利用哈希表来进行查找满足条件的值。有如下几种分组方式:
- 计算
a+b
,放入哈希表,然后遍历nums3+nums4,在哈希表中查找0-(c+d)
(a、b、c、d分别为四个数组中元素),时间复杂度O(n^2); - 计算
a+b+c
放入哈希表,然后遍历nums4,在哈希表中查找0-d
,时间复杂度O(n^3)。
由上可以看出,采用两两分组的时间复杂度相对较低。
- 计算
代码
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int, int> umap;
for (int a : nums1) {
for (int b : nums2) {
umap[a+b]++;
}
}
int count = 0;
for (int c : nums3) {
for (int d : nums4) {
int target = 0 - (c + d);
if (umap.find(target) != umap.end()) {
count += umap[target]; //注意加的是target出现的次数
}
}
}
return count;
}
};
383. 赎金信
题目链接:383. 赎金信
思路
这道题类似前面的字母异位词,同样是看两个字符串是否匹配问题,可以用相同的思路来解。
- 首先比较两个字符串长度,如果ransomNote的长度比magazine的长,肯定不满足
- 然后首先遍历magazine,将字母出现的次数记录在一个大小为26的数组中
- 遍历ransomNote,将数组中对应字母次数减掉
- 如果数组中元素出现小于0的情况,证明ransomNote中此字母个数多,不满足要求
代码
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
int record[26] = {0};
if (ransomNote.size() > magazine.size()) {
return false;
}
for (char ch : magazine) {
record[ch-'a']++;
}
for (char ch : ransomNote) {
record[ch-'a']--;
if(record[ch-'a'] < 0) {
return false;
}
}
return true;
}
};
15. 三数之和
题目链接:15. 三数之和
思路
此题可以考虑类似两数之和的哈希法,但是题目要求不重复,而且很多小细节很难处理,因此双指针法是比较好的选择。
首先对数组排序,定义i遍历数组,然后设两个指针left = i + 1
,right = nums.size() - 1
,判断三个指针所指元素之和,如果大于0,证明right大了,向左移动;如果小于0,证明left小了,向右移动;等于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;
}
//对i去重,因为前面已经遍历过所有的可能性
//注意去重的细节,下面这种将漏掉类似[-1,-1,2]这种情况
/*
if(nums[i]=nums[i+1]) {
continue;
}
*/
//正确去重方法
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[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]});
//对right和left去重,应该放在找到第一个三元组之后,防止漏掉[0,0,0]的情况。
while (right > left && nums[right] == nums[right - 1]) right--;
while (right > left && nums[left] == nums[left + 1]) left++;
//找到答案时,双指针同时收缩
right--;
left++;
}
}
}
return result;
}
};
18. 四数之和
题目链接:18. 四数之和
思路
此题和四数相加还是有区别的,四数相加是在四个数组,而此题是在一个数组,和三叔之和相同,同样运用双指针法,只是多了一层循环
- 注意剪枝的条件,此题要求和为target,考虑负数的情况
- 注意去重的条件
代码
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;
}
//对k去重
if (k > 0 && nums[k] == nums[k-1]) {
continue;
}
//下一层循环和三数之和代码类似,注意i的起始位置
for (int i = k + 1; i < nums.size(); i++) {
if (nums[k] + nums[i] > target && nums[k] + nums[i] >= 0) {
break;
}
if (i > k+1 && nums[i] == nums[i-1]) {
continue;
}
int left = i + 1;
int right = nums.size() - 1;
while (right > left) {
if ((long) nums[k] + nums[i] + nums[left] + nums[right] > target) right--;
else if ((long) nums[k] + nums[i] + nums[left] + nums[right] < target) left++;
else {
result.push_back(vector<int>{nums[k], 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 result;
}
};
总结
今天这几道题目还是值得深思的,好好思考一下双指针操作的精妙之处,注意剪枝和去重的细节。