四数相加
注意📍
预处理前两个数组,把a+b 放入key,并统计出现过的次数value,再遍历后两个数组,判断0-(c+d)有没有在这个集合中出现过。
如果只遍历第一个数组,去查后三个,时间复杂度是O(n^3)
class Solution {
public:
int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
unordered_map<int,int> umap; // key:a+b的数值 value:a+b数值的出现次数
for(int a : A) {
for(int b : B) {
umap[a + b]++;
}
}
int count = 0; // 统计a+b+c+d =0出现的次数
//遍历大C和大D数组,如果找到0-(c+d)在map中出现过的话,就把map中key对应的value统计出来
for(int c : C) {
for(int d : D) {
if(umap.find(0 -(c + d)) != umap.end()) {
count += umap[0 -(c + d)];
}
}
}
return count;
}
};
赎金信
//暴力解法
//时间复杂度:O(n^2)
//空间复杂度:O(1)
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
for(int i = 0; i < magazine.length(); i++) {
for(int j = 0; j < ransomNote.length(); j++) {
if(magazine[i] == ransomNote[j]) {
ransomNote.erase(ransomNote.begin() + j); //ransomNote删除这个字符
break;
}
}
}
//如果ransomNote为空,则说明magazine的字符可以组成ransomNote
if(ransomNote.length() == 0) {
return true;
}
return false;
}
};
//哈希解法
//时间复杂度:O(n)
//空间复杂度:O(1)
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
int record[26] = {0};
if(ransomNote.size() > magazine.size()) {
return false;
}
for(int i = 0; i < magazine.length(); i++) {
record[magazine[i]-'a']++; // 记录magazine里各个字符出现次数
}
for(int j = 0; j < ransomNote.length(); j++) {
record[ransomNote[j]-'a']--;
// 如果小于零说明ransomNote里出现的字符,magazine没有
if(record[ransomNote[j]-'a'] < 0) {
return false;
}
}
return true;
}
};
三数之和
注意📍
双指针法,去重是关键;
对数组要进行排序,一层for循环num[i]为确定值,然后循环内有left和right下标作为双指针,找到nums[i]+nums[left]+nums[right]==0;
如果三数相加大于零,right--,相加小于零,left++,等于零的话存入result数组
区分nums[i]==nums[i+1]和nums[i]==nums[i-1],nums[i]==nums[i+1],会排除掉三元组内元素数值相等的情况,比如[-1,-1,2]
right和left不能相等,相等的话只有两个数,不符合三元组
去重一定要放在获取到一个三元组之后
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) { // 排序之后,第一个大于0,必然没有满足条件的三元组
return result;
}
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]});
// 去重逻辑应该放在找到一个三元组之后,对b和c去重
while(right > left && nums[right] == nums[right - 1]) right --;
while(right > left && nums[left] == nums[left + 1]) left ++;
right--;
left++;
}
}
}
return result;
}
};
四数之和
注意📍
不能直接判断nums[k] >target就返回了,比如[-3,-2,0,0] ,target为-5,-3>-5,这样就去掉了符合的四元组
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;
}
// 对nums[k]进行去重
if(k > 0 && nums[k] == nums[k - 1]) {
continue;
}
for (int i = k + 1; i < nums.size() ;i++) {
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) {
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]});
// 对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;
}
};