哈希表
四数相加II
这道题一共给了四个数组,让返回满足题意的元组的个数。
大家的第一反应就是用四重循环挨个遍历,不出意外的话会超出时间限制,我们不妨把四个数组拆开,拆成两个和两个。
也就是说我们可以统计第一个和第二个数组各个元素之和sum,然后记录下来,接着统计第三个和第四个数组各个元素之和sum2,然后检查sum与sum2是否为相反数。(这里的sum和sum2是帮助理解的,在下面代码中并没有出现)
要记录一二个数组之和,我们就要用到哈希表,但到底是用set还是map呢?
set也就是数学上的集合,其中的元素都是不重复的,例如,数组一的第一个元素和第二个元素相等,那么在统计和的时候也会遇到sum相等的情况,这个时候用set是无法记录sum出现的次数的。
map就可以完成这个任务,比如和为 4 的情况出现了两次,那么map中key为 4 的时候所对应的value就是 2 ,我们让最后输出的count加上value的值就好了。
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;
}
};
赎金信
这道题和有效字母异位词基本上没有区别,就不过多赘述了。
这里也给大家拓展一下字符串和数组有什么差别,
字符串是若干字符组成的有限序列,也可以理解为是一个字符数组,但是很多语言对字符串做了特殊的规定,接下来我来说一说C/C++中的字符串。
在C语言中,把一个字符串存入一个数组时,也把结束符 '\0’存入数组,并以此作为该字符串是否结束的标志。
那么vector< char > 和 string 又有什么区别呢?
其实在基本操作上没有区别,但是 string提供更多的字符串处理的相关接口,例如string 重载了+,而vector却没有。
所以想处理字符串,我们还是会定义一个string类型。
纯搬运,因为我也不知道
三数之和
这道题用哈希表的话会比较麻烦,咱们直接看双指针法吧。
这道题给了我们三个数组,并且我们需要去遍历,如果用三重循环的话这道题就没有意义了,那我们可以根据上一道题的思路,把三个数组拆成一和二,也就是说我们只需要两个循环来完成三个循环所要做的事情
那么我们就需要借助双指针了。由于这道题目只是跟元素有关,跟数组元素的下标没有关系,所以我们可以先对数组进行排序,这样能够方便我们进行遍历。
第一层循环我们设置指针i,第二层设置left和right两个指针。这样我们就可以同时遍历三个数了。第一层我们不妨用for循环,内层需要while循环,条件就是 left < right
我们让i指向最开始的元素,left指向i紧跟着的后面的元素,right指向最右边的元素,然后让left和right移动,寻找满足条件的元素。
int left = i+1;
int right = nums.size()-1;
因为我们已经进行了排序,所以左边的元素一定小于等于右边的元素,所以我们可以根据三数之和进行判断,如果nums[i],nums[left],nums[right]的和大于0,说明right指向的元素过大,我们需要将right指针向左移动。
那为啥不让left向左移动嘞?因为left是紧挨着i的,没有办法向左移动。如果和小于0,说明left指向的值太小了,需要向右移动。
换句话说,i的作用是设置一个基准,left和right只能在基准的右边开始寻找,left是在和过小时,向右移动,让值变大;right是当值过大时,向左移动,让值变小的。
重复这个过程,就可以找到和为0的情况(存在这种情况的话)。
内层循环就是上面的逻辑
if(nums[i] + nums[left]+ nums[right] > 0) right -- ;
else if(nums[i] + nums[left] + nums[right] < 0) left ++;
这只是内循环的一部分
如果找到了符合的情况,我们就放到结果集里面,当然,我们还得面临重复的问题。
举个例子 {-1,-1,2,3} 这个数组,一开始让i指向最左边的-1,right和left遍历完之后i会向右移动,但是i指向的还是 -1,那么在i没有向右移动的时候,其中一种情况是 -1,2,3 并且在i指向第二个-1的时候,会有一种情况跟上面提到的完全一样,(也就是i还是-1,left还是2,right还是3)那么这就是重复。这只是一个例子,帮助理解
在我们找到结果之后就需要去重,对于left和right的去重操作其实比较好想
while(right > left && nums[right] == nums[right - 1]) right--;
while(right > left && nums[left] == nums[left + 1]) left++;
也就是如果连着多个元素是一样的,直接跳过就好了。
为什么找到之后才去重呢?可以这么想,如果找到一个结果,就要对它保驾护航,不能让重复的情况对这个结果产生影响。当然,更有说服力的例子在最底下的完整代码当中。
while内的代码如下
if(nums[i] + nums[left]+ nums[right] > 0) right -- ;
else if(nums[i] + nums[left] + nums[right] < 0) left ++;
else{
rst.push_back(vector<int>{nums[i],nums[left],nums[right]});//rst就是结果集
while(right > left && nums[right] == nums[right - 1]) right--;
while(right > left && nums[left] == nums[left+1]) left++;
right --;
left ++;
}
还有就是对i去重了,那判断重复的时候是看i的前一个跟i一样,还是后一个呢?
这时候就可以举一些特殊的例子了,以特殊的三元数组来说,nums为[-1,-1,2]这种情况是符合题意的,那么我们可以带进去看一下哪一种去重方式符合,不难发现,是第一种,也就是看i的前一个。
我们把对i去重的操作放到前面就行。
完整代码如下
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;
}
};
就先写到这里吧。