@TOC
一.map容器
map是一种key value的存储结构,可以用key保存数值,用value再保存数值所在的下标。
c++标准库有三种形式。
映射 | 底层实现 | 是否有序 | 数值是否可以重复 | 能否更改数值 | 查询效率 | 增删效率 |
---|---|---|---|---|---|---|
std::map | 红黑树 | key有序 | key不可重复 | key不可修改 | O(log n) | O(log n) |
std::multimap | 红黑树 | key有序 | key可重复 | key不可修改 | O(log n) | O(log n) |
std::unordered_map | 哈希表 | key无序 | key不可重复 | key不可修改 | O(1) | O(1) |
二.两数求和
1.题目描述
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
输入:nums = [3,2,4], target = 6
输出:[1,2]
示例 3:
输入:nums = [3,3], target = 6
输出:[0,1]
2.思路
当我们需要查询一个元素是否出现过,或者一个元素是否在集合里的时候,就要第一时间想到哈希法,该题需要一个集合来存放我们遍历过的元素,然后在遍历数组的时候去询问这个集合,某元素是否遍历过,也就是是否出现在这个集合。相比暴力求解,该方法的时间复杂度为时间复杂度: O(n)。
3.代码
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int>map;//创建哈希表
for(int i=0;i<nums.size();i++)
{
auto iter=map.find(target-nums[i]);//差值
if(iter!=map.end())//找到匹配的数值
{
return{i,iter->second};//返回两个数的标志位
}
map.insert(pair<int,int>(nums[i],i));//添加数值和其位置
}
return {};
}
};
三.四数相加
1.题目描述
给你四个整数数组 nums1、nums2、nums3 和 nums4 ,数组长度都是 n ,请你计算有多少个元组 (i, j, k, l) 能满足:
0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
示例 1:
输入:nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
输出:2
解释:
两个元组如下:
- (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
- (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0
2.思路
该题类似于两数求和,只是target=0,不同点在于该题需要返回的是满足条件的元组的总个数,不需要返回元素的下标。
我们可以将四个数前两个和后两个分别求和,这样就将该题转换为两数求和,注意点为不同的两个数的和可能一样,由于只需要满足条件的数组个数,所以可以采用哈希表来key保存两个数的和,value保存满足该和的个数。
具体步骤如下
1.创建map,遍历nums1和nums2,将两两数据的和保存到map中,其中key值为和,value值为次数。
2.定义count为满足条件的数组个数,遍历num3和num4,查询map中是否有满足0-nums3[i]-nums4[j]的元素,存在表示满足条件,count加该元素对应得value值。
3.遍历结束,返回count。
3.代码
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int,int>map;
for(int i=0;i<nums1.size();i++)
for(int j=0;j<nums2.size();j++)
map[nums1[i]+nums2[j]]++;//数值对于的value增加1
int count=0;
for(int i=0;i<nums3.size();i++)
for(int j=0;j<nums4.size();j++)
{
int temp=nums3[i]+nums4[j];
if(map.find((0-temp))!=map.end())
count+=map[(0-temp)];//count+map中找到的对应值的次数
}
return count;
}