454.四数相加II
分组+哈希表:
将四个数组分成两部分,A 和 B 为一组,C 和 D 为另外一组。
对于 A 和 B,我们使用二重循环对它们进行遍历,得到所有 A[i]+B[j] 的值存入哈希映射中,每个键表示A[i]+B[j]的和,对应的值为出现的次数。
对于 C 和 D,我们同样使用二重循环对它们进行遍历。当遍历到 C[k]+D[l] 时,如果 -(C[k]+D[l])出现在哈希映射中,那么将 -(C[k]+D[l])对应的值累加进答案中。
最终即可得到满足 A[i]+B[j]+C[k]+D[l]=0的四元组数目。
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int, int>map;//a+b,出现的次数
int res = 0;
for(int a : nums1){
for(int b : nums2){
map[a+b]++;
}
}
for(int c : nums3){
for(int d : nums4){
int target = -(c+d);
if(map.find(target) != map.end())
res += map[target];
}
}
return res;
}
};
383.赎金信
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
int hash[26] = {0};
if(ransomNote.length() > magazine.length())
return false;
for(char a : magazine)
hash[a-'a']++;
for(char a : ransomNote)
if(--hash[a-'a'] < 0)
return false;
return true;
}
};
15.三数之和
双指针法:先对数组排序(因为不要求返回下标),然后双指针移动,难点在于对三元组{a,b,c}中a,b,c的去重。
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()-2; i++){
if(nums[i] > 0)//第一个数大于0,不可能满足和为0
break;
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{//满足条件,加入结果集中
res.push_back(vector<int>{nums[i], nums[left], nums[right]});
left++,right--;//收缩边界
while(left < right && nums[left] == nums[left-1]) left++;//对第二,三个数去重
while(left < right && nums[right] == nums[right+1]) right--;
}
}
}
return res;
}
};
18.四数之和
思路与三数之和相似,使用双指针法,在三数之和的基础上再套一层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;
//nums[i]去重
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;
//nums[j]去重
if(j > i + 1 && nums[j] == nums[j-1]) continue;
int left = j + 1;
int right = nums.size()-1;
while(left < right){
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]});
left++,right--;//收缩边界
while(left < right && nums[left] == nums[left-1]) left++;//对第二,三个数去重
while(left < right && nums[right] == nums[right+1]) right--;
}
}
}
}
return res;
}
};