学习目标:
● 454.四数相加II
● 383. 赎金信
● 15. 三数之和
● 18. 四数之和
● 总结
学习内容:
- 四数相加II
- 赎金信:map比数组空间消耗大
- 三数之和 只有一个数组,不好去重,因此用多指针
- 四数之和
学习时间:
2022.11.1
学习产出:
454.四数相加II
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int,int> umap;
for(int i =0;i<nums1.size();i++){
for(int j = 0;j<nums2.size();j++){
umap[nums1[i]+nums2[j]]++;
}
}
int count=0;
for(int m =0;m<nums3.size();m++){
for(int n = 0;n<nums4.size();n++){
if(umap.find(0-(nums3[m]+nums4[n]))!=umap.end()){
count+= umap[0-(nums3[m]+nums4[n])];
}
}
}
return count;
}
};
赎金信
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
unordered_map<char,int> umap;
for(char ch:magazine){
umap[ch]++;
}
for(char ch2:ransomNote){
auto iter = umap.find(ch2);
if(iter==umap.end()) return false;
if(iter->second!=0){
umap[ch2]--;
}
else return false;
}
return true;
}
};
// 时间复杂度: O(n)
// 空间复杂度:O(1)
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
int record[26] = {0};
//add
if (ransomNote.size() > magazine.size()) {
return false;
}
for (int i = 0; i < magazine.length(); i++) {
// 通过recode数据记录 magazine里各个字符出现次数
record[magazine[i]-'a'] ++;
}
for (int j = 0; j < ransomNote.length(); j++) {
// 遍历ransomNote,在record里对应的字符个数做--操作
record[ransomNote[j]-'a']--;
// 如果小于零说明ransomNote里出现的字符,magazine没有
if(record[ransomNote[j]-'a'] < 0) {
return false;
}
}
return true;
}
};
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(nums[i]>0) return res;
if(i>0&&nums[i]==nums[i-1]) continue;
int lft = i+1;
int right = nums.size()-1;
while(lft<right){
if(nums[i]+nums[lft]+nums[right]>0){
right--;
}
else if(nums[i]+nums[lft]+nums[right]<0){
lft++;
}
else{
res.push_back(vector<int>{nums[i],nums[lft],nums[right]});
while(right>lft&&nums[right]==nums[right-1]) right--;
while(right>lft&&nums[lft]==nums[lft+1]) lft++;
lft++;
right--;
}
}
}
return res;
}
};
四数之和
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; // 这里使用break,统一通过最后的return返回
}
// 对nums[k]去重
if (k > 0 && nums[k] == nums[k - 1]) {
continue;
}
for (int i = k + 1; i < nums.size(); i++) {
// 2级剪枝处理
if (nums[k] + nums[i] > target && nums[k] + nums[i] >= 0) {
break;
}
// 对nums[i]去重
if (i > k + 1 && nums[i] == nums[i - 1]) {
continue;
}
int left = i + 1;
int right = nums.size() - 1;
while (right > left) {
// nums[k] + nums[i] + nums[left] + nums[right] > target 会溢出
if ((long) nums[k] + nums[i] + nums[left] + nums[right] > target) {
right--;
// nums[k] + nums[i] + nums[left] + nums[right] < target 会溢出
} 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]});
// 对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;
}
};