D7: 焦虑😐
454. 4Sum II
两数之和,正常要
n
4
n^4
n4复杂度,使用map可以降低到
n
2
n^2
n2复杂度。先创建map保存a+b的和,再创建map保存c+d的和,然后在第二个map中找
0
−
(
a
+
b
)
0-(a+b)
0−(a+b)即可。代码如下:
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int, int> sumsAB, sumsCD;
for(int a : nums1) {
for(int b : nums2) {
sumsAB[a + b]++;
}
}
for(int c : nums3) {
for(int d : nums4) {
sumsCD[c + d]++;
}
}
int count = 0;
for (const auto& pair : sumsAB) {
int complement = -pair.first; //相当于是0-(a+b)
if (sumsCD.find(complement) != sumsCD.end()) {
count += pair.second * sumsCD[complement];
}
}
return count;
}
};
383. Ransom Note
一开始想着无脑用map,看了Carl解释才知道在本题的情况下,使用map的空间消耗要比数组大一些的,因为map要维护红黑树或者哈希表,而且还要做哈希函数,是费时的!数据量大的话就能体现出来差别了。 所以数组更加简单直接有效!
代码如下:
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;
}
};
15. 3Sum
这个确实哈希表有困难,双指针法就简单了,固定数组元素i,左指针指向i+1个元素,右指针指向数组尾部,然后根据左右指针之和来和-i比较,再移动左指针或者右指针
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
set<vector<int>> result;
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size(); ++i) {
// 跳过重复的数字
if (i > 0 && nums[i] == nums[i-1]) continue;
int target = -nums[i];
int left = i + 1;
int right = nums.size() - 1;
while (left < right) {
int sum = nums[left] + nums[right];
if (sum == target) {
result.insert({nums[i], nums[left], nums[right]});
// 移动指针,跳过重复的数字
while (left < right && nums[left] == nums[left + 1]) left++;
while (left < right && nums[right] == nums[right - 1]) right--;
left++;
right--;
} else if (sum < target) {
left++;
} else {
right--;
}
}
}
return vector<vector<int>>(result.begin(), result.end());
}
};
- 4Sum
四数之和,区别点在于,这四个数字在一个数组中。同样使用双指针法,重点是剪枝操作。代码如下:
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;
}
};
总结:哈希表告一段落,重点是如何理解三种不同的结构的适用场景,
数组,set,map各有特点。
-
数组的大小是有限的,受到系统栈空间(不是数据结构的栈)的限制。
-
如果数组空间够大,但哈希值比较少、特别分散、跨度非常大,使用数组就造成空间的极大浪费。
-
set是一个集合,里面放的元素只能是一个key,而两数之和这道题目,不仅要判断y是否存在而且还要记录y的下标位置,因为要返回x 和 y的下标。所以set 也不能用。
-
要在恰当的地方使用双指针法。