四数相加 ||
例题454:
给你四个整数数组 nums1、nums2、nums3 和 nums4 ,数组长度都是 n ,请你计算有多少个元组 (i, j, k, l) 能满足:
- 0 <= i, j, k, l < n
- nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
/*用一个map存入1和2数组的和以及和出现的次数,用count来存1 + 2 + 3 + 4 = 0的次数
然后遍历3和4数组,找到是否有1+2和的负数,如果有的话就用count将map中的value取出来也就是次数*/
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
int count = 0;
unordered_map<int, int> hmap;
int i, j,sum1,sum2;
for (i = 0; i < nums1.size(); i++)
{
for (j = 0; j < nums2.size(); j++)
{
sum1 = nums1[i] + nums2[j];
hmap[sum1]++;
}
}
for (i = 0; i < nums3.size(); i++)
{
for (j = 0; j < nums4.size(); j++)
{
sum2 = nums3[i] + nums4[j];
auto it = hmap.find(0 - sum2);
if (it != hmap.end())
{
count += it->second;
}
}
}
return count;
}
};
此题是在四个数组容器里各取一个数,比一个数组取四个数和为0更容易
赎金信
例题383:给你两个字符串:ransomNote 和 magazine ,判断 ransomNote 能不能由 magazine 里面的字符构成。
如果可以,返回 true ;否则返回 false 。
magazine 中的每个字符只能在 ransomNote 中使用一次。
bool canConstruct(string ransomNote, string magazine) {
unordered_map<int, int> hmap(magazine.begin(), magazine.end());
for (char ch : ransomNote)
{
magazine[ch - 'a']--;
if (magazine[ch-'a']<0)
{
return false;
}
}
return true;
}
三数之和(较难)
例题15:给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
该题目较难的点在于怎么去重,答案中不能存在相同的三元组,例如(-1,0,1,1)首先取(-1,0,1)这个三元组,如果1向后移一位又取(-1,0,1)重复。如果暴力法先找到所有相加为0的三位数,不能保证它们不重复,因此还需要哈希表来去重,此时时空复杂度都较高,并且去重要写的判断较多,一下不容易写全写对,因此,此题不适合用哈希表。
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
//下标可以不同,但数字必须不同,如果数字相同只能算一种情况
vector<vector<int>> res;
int i;
sort(nums.begin(), nums.end());//双指针方法要排序
for (i = 0; i < nums.size(); i++)
{
if (nums[i] > 0)
{
return res;//如果第一个数大于0,排序后不存在三元组和为0
}
if (i > 0 && nums[i] == nums[i - 1])//i>0要写在前面,如果写在后面i-1当i=0时有问题
{
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]});//push_back需要注意推进去的是什么类型,如果是普通容器则推一个数字或字符,如果是二维容器,推进去的就是一维容器
//如果nums[i]=4,后有两组-2,-2的组合,这相当于是一种组合,需要对left和right去重
while (right > left && nums[left] == nums[left + 1]) left++;
while (right > left && nums[right] == nums[right - 1]) right--;
//找到三元组之后left和right移动
left++;
right--;
}
}
}
return res;
}
};
四数之和
例题18:给你一个由 n 个整数组成的数组 nums ,和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] (若两个四元组元素一一对应,则认为两个四元组重复):
0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按 任意顺序 返回答案 。
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> res;
sort(nums.begin(), nums.end());
//类似于三数之和,不过第一个存放的前两个数之和
int i, j, left, right;
for (i = 0; i < nums.size() ; i++)
{
//对i剪枝
if(nums[i]>target && nums[i]>0)
{
break;
}
//对i去重
if(i>0 && nums[i]==nums[i-1])
{
continue;
}
for(j=i+1 ;j<nums.size();j++)
{
//对j去重
if(j>i+1 && nums[j]==nums[j-1])//注意j的范围是大于i+1,而不是1,如果是大于1,当i和j相同时会跳过
{
continue;
}
//对j剪枝
if(nums[i]+nums[j]>target && nums[i]+nums[j]>0)
{
break;
}
left = j + 1;
right = nums.size() - 1;
while (right > left)
{
if ((long)nums[i] + nums[j] + nums[left] + nums[right] > target)//注意四数相加可能超出int的取值范围,需要向上强转
{
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]});
while (right > left && nums[right] == nums[right - 1])
{
right--;
}
while (right > left && nums[left] == nums[left + 1])
{
left++;
}
right--;
left++;
}
}
}
}
return res;
}
};
该题需要注意:
1.与三数之和类似,不过前面固定的数变为了第一个和第二个数的和,需要对四个数分别进行去重
2.第二个数去重的范围是大于第一个数的下一位
3.四个数相加可能大于int的取值范围会溢出,需要向上强制转型为long
4.对第一个数剪枝,注意nums[i]>0,因为是负数时,满足条件
总结
- 一般来说哈希表是用来快速判断某个元素是否出现在一个集合中
- 清楚哈希函数和哈希碰撞在哈希表中的作用
- 哈希函数是将传入的key映射到索引上
- 哈希碰撞是指有多个key映射到相同的索引上的情况,可以用拉链法(后接链表)和线性探测法(后空索引地址存放碰撞元素)解决
- 哈希表有三种数据结构,分别是数组、set、map,需要了解它们各自的底层逻辑,以及各自适用的场景
数组一般有大小限制,且方便索引与key之间的映射,map需要哈希函数转换并且空间占用更大
没有大小限制的时候无法用数组,且索引分散跨度大,可以考虑set。