算法训练营 第七天
哈希day2~
454. 四数之和II
这道题的暴力解法是四层循环，但是时间复杂度太高了o(n^4)，我想的方法是先遍历两个数组，把他们两数相加之和存进哈希表，键值对是两数相加的值和该数出现的次数，然后再遍历剩下两个数组。
我的源码
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int, int>hash;
hash[1]++;
hash[2];
cout << hash[1] << endl;
cout << hash[2] << endl;
int count = 0;
for (auto a : nums1) {
for (auto b : nums2) {
if (hash.find(a + b) != hash.end()) {
hash[a + b]++;
}
else
hash.insert(make_pair(a + b, 1));
}
}
for (auto a : nums3) {
for (auto b : nums4) {
if (hash.find(-a - b) != hash.end()) {
count += hash[-a - b];
}
}
}
return count;
}
};
参考答案
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数值出现的次数
// 遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中
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;
}
};
解题思路和参考答案一样，这道题我之前刷过，因为只用统计四数相加=0的数量，不用返回下标值，因此用哈希表记录值出现的次数即可。
383. 救赎金
这道题的解题思路和之前的哈希表题一样，是一看就知道要用哈希法的题，先用哈希遍历magazine存进去，然后再遍历ransomNote，如果存在hash[a]<0则返回false；
我的源码
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
unordered_map<char, int>hash;
for (auto a : magazine)
hash[a]++;
for (auto b : ransomNote) {
if (hash.find(b) == hash.end())
return false;
hash[b]--;
if (hash[b] < 0)
return false;
}
return true;
}
};
参考答案
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++) {
// 通过record数据记录 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;
}
};
这道题我没想到直接用数组存就行，毕竟要存的key是字母，一共就26个字母，因为用的unordered_map，导致我的答案运行时间和内存都要略高一些
15. 三数之和
这道题，想到的第一个方法是暴力解法，直接三层for循环，但是这个方法太不优雅了，作为一道哈希题，想想怎么能够用上哈希，emmmm，想不出来，看答案！
看完了，回来说说感受，这道题重点在减枝，哈希表不好减枝（因为哈希表是无序的），所以用双指针法更合适（排序了之后，重复的元素会挨在一起，更好减枝）。对于三个数a,b,c，卡尔对b和c的剪枝是放在了while循环之后。我觉得放在while循环之前也行，因为我觉得nums[left]应该和nums[left-1]比较才对，nums[left-1]是已经处理过的数，如果left和left-1一样就不用再处理了，我试了试这个思路，可以AC成功，代码如下：
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(), nums.end());
vector<vector<int>> res;
for (int i = 0; i < nums.size() - 2; i++) {
if (nums[i] > 0)
break;
if(i > 0 && nums[i] == nums[i - 1])
continue;
int left = i + 1;
int right = nums.size()-1;
while (right > left) {
while (left > i + 1 && right > left && nums[left] == nums[left - 1])
left++;
while (right<nums.size() - 1 && right>left && nums[right] == nums[right + 1])
right--;
if (right <= left)
break;
int tmp = nums[i] + nums[left] + nums[right];
if (tmp < 0)
left++;
else if (tmp > 0)
right--;
else {
res.push_back({ nums[i],nums[left],nums[right] });
left++;
right--;
}
}
}
return res;
}
};
不过res.push_back之后，少不了要对left和right进行处理，在这个时候剪枝也合理，也就是参考答案的做法。
参考答案
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> result;
sort(nums.begin(), nums.end());
// 找出a + b + c = 0
// a = nums[i], b = nums[left], c = nums[right]
for (int i = 0; i < nums.size(); i++) {
// 排序之后如果第一个元素已经大于零，那么无论如何组合都不可能凑成三元组，直接返回结果就可以了
if (nums[i] > 0) {
return result;
}
// 错误去重a方法，将会漏掉-1,-1,2 这种情况
/*
if (nums[i] == nums[i + 1]) {
continue;
}
*/
// 正确去重a方法
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
int left = i + 1;
int right = nums.size() - 1;
while (right > left) {
// 去重复逻辑如果放在这里，0，0，0 的情况，可能直接导致 right<=left 了，从而漏掉了 0,0,0 这种三元组
/*
while (right > left && nums[right] == nums[right - 1]) right--;
while (right > left && nums[left] == nums[left + 1]) 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;
}
};
18. 四数之和
这道题和四数之和II很像，但这个要返回具体的值，感觉还得用排序，但没有具体的思路，看看答案
看完答案发现是自己懒惰了（三数之和的逻辑有些复杂，懒得想），其实和三数之和的思路一样，就是再多一层循环，知道解题思路了之后自己写了一下
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>>res;
if (nums.size() < 4)
return res;
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size() - 3; i++) {
if ((target >= 0||nums[i]>0) && nums[i] > target)
break;
if (i > 0 && nums[i] == nums[i - 1])
continue;
for (int j = i + 1; j < nums.size() - 2; j++) {
int tmp = target - nums[i];
if ((tmp >= 0 || nums[j] > 0) && nums[j] > tmp)
break;
if (j > i+1 && nums[j] == nums[j - 1])
continue;
int left = j + 1;
int right = nums.size() - 1;
while (right > left) {
auto sum = (long)nums[i] + nums[j] + nums[left] + nums[right];
if (sum > target)
right--;
else if (sum < target)
left++;
else {
res.push_back({ nums[i], nums[j], nums[left], nums[right] });
while (right > left && nums[left] == nums[left + 1])
left++;
while (right > left && nums[right] == nums[right - 1])
right--;
left++;
right--;
}
}
}
}
return res;
}
};
要注意的是，四数相加可能超过int的取值范围，得用long来存储。写的和参考答案基本一摸一样，所以参考答案我就不放了。
今日任务完成！去吃饭！
906
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
