哈希表二
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中,用key记录加和,value记录出现次数。
在遍历c+d,查找0-(c+d)在map中出现的次数,对次数加和即可得到元组个数。
代码实现
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int,int>m1;
//统计a+b出现的次数,放入m1中
for (int a:nums1)
{
for(int b:nums2)
{
m1[a+b]++;
}
}
int count=0;//统计a+b+c+d=0出现的次数
//遍历c、d数组,统计0-(c+d)在map中出现的次数
for(int c:nums3)
{
for(int d:nums4)
{
if(m1.find(0-c-d)!=m1.end())//如果在m1中找到0-c-d对应的键值
{
count+=m1[0-(c+d)];//将键值对应的value即出现次数加到结果中
}
}
}
return count;//返回可行方案的个数
}
};
总结
383.赎金信
题目描述
给你两个字符串:ransomNote
和 magazine
,判断 ransomNote
能不能由 magazine
里面的字符构成。
如果可以,返回 true
;否则返回 false
。
magazine
中的每个字符只能在 ransomNote
中使用一次。
示例 1:
输入:ransomNote = "a", magazine = "b"
输出:false
解题思路
代码实现
class Solution {
public:
bool canConstruct(string ransomNote, string magazine) {
//利用哈希表存字母,magazine中的字母使表中value++,ransomNote中的字母使表--
//最后判断表中的value是否有小于0的,若有则为false。
vector<int> result(26,0);
for(char i:magazine){
result[i-'a']++;
}
for(char j:ransomNote)
{
result[j-'a']--;
}
for(int i=0;i<26;i++)
{
if (result[i]<0)
{
return false;
}
}
return true;
}
};
总结
未看讲解,按自己思路写的,但是没有一遍ac。出bug的点在于vector初始化又写错了
切记,一维vector初始化按照如下写法,用小括号,因为调用了构造函数
vector<int> result(26,0);
二维vector定义及初始化按如下写法
//第一种写法,集成式
vector<vector<int>>res(5,vector<int>{1,2,3,4})
//第二种写法
vector<int>v1{1,2,3,4};
vector<vector<int>>res(5, v1);
15 三数之和
题目描述
给你一个整数数组 nums
,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j
、i != k
且 j != k
,同时还满足 nums[i] + nums[j] + nums[k] == 0
。请
你返回所有和为 0
且不重复的三元组。
**注意:**答案中不可以包含重复的三元组。
解题思路
双指针。先对元素排序。对第一个元素遍历,第二个 元素和第三个元素分别利用左指针和右指针去查找。通过求解三元素的和与0的大小关系来判断左右指针移动方向。
注意:三个元素均需要进行去重操作。对遍历的元素需要进行剪枝操作。
代码实现
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>>result;
//先对原始数组排序
sort(nums.begin(),nums.end());
//用双指针法找a+b+c=0的元素
//用i来遍历数组,并记录a,然后用双指针记录b、c
for(int i=0;i<nums.size();i++)
{
if(nums[i]>0)//对于排过序的数组,若a已经大于0,说明三数无法加和为0
{
return result;
}
//对出现的重复数进行去重
if(i>0&&nums[i]==nums[i-1])
{
continue;
}
int left=i+1;
int right=nums.size()-1;
int j=0;
//开始用双指针找三个和为0的数
while(left<right){
//根据三数之和与0的关系开始移动两个指针
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]});
//j++;
//cout<<j<<endl;
//找到一个三元组以后再去重,对left和right指向的数字也去重
while(left<right&&nums[left]==nums[left+1])
{
left++;
}
while(left<right&&nums[right-1]==nums[right])
{
right--;
}
//更新双指针边界
left++;
right--;
}
}
}
return result;
}
};
总结
注意:三个元素均需要进行去重操作。对遍历的元素需要进行剪枝操作。
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
你可以按 任意顺序 返回答案 。
示例 1:
输入:nums = [1,0,-1,0,-2,2], target = 0
输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
解题思路
在三数之和的基础上再套一层循环。
代码实现
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返回
}
//对k去重
if(k>0&&nums[k]==nums[k-1])
{
continue;
}
//两层for循环遍历nums前两个数加和值作为第一个指针,
for(int i=k+1;i<nums.size();i++)
{
// 2级剪枝处理
if (nums[k] + nums[i] > target && nums[k] + nums[i] >= 0) {
break;
}
//对i去重
if(i>k+1&&nums[i]==nums[i-1])
{
continue;
}
int left=i+1;
int right=nums.size()-1;
while(left<right)
{
if(nums[k]+nums[i]>(target-(nums[left]+nums[right])))
{
right--;
}
else if(nums[k]+nums[i]<(target-(nums[left]+nums[right])))
{
left++;
}
else{
result.push_back(vector<int>{nums[k],nums[i],nums[left],nums[right]});
//找到四元组后对reght去重
while(right>left&&nums[right]==nums[right-1])
{
right--;
}
//对left去重
while(right>left&&nums[left]==nums[left+1])
{
left++;
}
//找到答案,同时移动指针
left++;
right--;
}
}
}
}
return result;
}
};
总结
<代码随想录>的书本上代码少了剪枝的操作,在遇到nums中的数组过大,接近int上限时代码会出错。
剪枝是针对遍历的元素进行的,因而再两层for循环中均需要剪枝。
去重是针对四个数的,因而在每一层循环内,以及left和right指针,这四个指向的数据都要先去重,才判断加和。