本文讲解两数之和,三数之和和四数之和相关的5道经典双指针类型题。会在详解题目的同时给出AC代码【这五个题目从前往后是循序渐进的,非常巧妙】
目录
1、查找总价格为目标值的两个商品
【本题即两数之和经典例题】
解法一:暴力解法
本题是升序的数组,因此可以用「对撞指针」优化时间复杂度。
算法流程(附带算法分析,为什么可以使用对撞指针):
a. 初始化 left , right 分别指向数组的左右两端(不是我们理解的指针,而是数组的下标)
b. 当 left < right 的时候,一直循环
- 当 nums[left] + nums[right] == target 时,说明找到结果,记录结果,并且返回;
- 当 nums[left] + nums[right] < target 时:
- 对于 nums[left] 而言,此时 nums[right] 相当于是 nums[left] 能碰到的最大值。如果此时不符合要求,我们可以让left++,使和变大
- 当 nums[left] + nums[right] > target 时,同理我们可以舍去nums[right] (因为和过大了,应该小一点)。让 right-- ,继续比较下一组数据,而 left 指针不变
class Solution {
public:
vector<int> twoSum(vector<int>& price, int target) {
int left = 0, right = price.size() - 1;
int sum = 0;
while (left < right)
{
sum = price[left] + price[right];
if (sum > target) right--;
else if (sum < target) left++;
else return {price[left], price[right]};//这么返回会被视为vector返回
}
//为了让编译器过,没结果也要有返回值
return {};//也可以随便设置一个返回值,比如return{-1, -1};
}
};
2、两数之和
本题与第一题区别在于:
- 1、数组不是升序的,就不能使用对撞指针,即使你排序后,对应下标也不是原来的下标,当然也可以哈希映射
- 2、返回值要返回下标,所以可以使用哈希映射
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> umap;//key:元素值 v:元素对应下标
for (int i = 0; i < nums.size(); i++){
// 遍历当前元素,并在map中寻找是否有匹配的key
int sum = target - nums[i];
if (umap.find(sum) != umap.end()){
return {i, umap[sum]};
}
// 如果没找到匹配对,就把访问过的元素和下标加入到map中
umap.insert(pair<int, int>(nums[i], i));
}
return {};
}
};
3、三数之和
找三元组的过程:
去重: 【当然也可以用set等,但是我们可以想另一个办法】
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> ret;
//1、排序
sort(nums.begin(), nums.end());
//2、双指针算法解决问题
int n = nums.size();
for (int i = 0; i < n;)
{
if (nums[i] > 0) break;//优化:若固定的数都>0,且是升序,后面之和不可能有负数
//沿用两数之和操作
int left = i + 1, right = n - 1 , target = -nums[i];
while (left < right)
{
int sum = nums[left] + nums[right];
if(sum < target) left++;
else if (sum > target) right--;
else{
ret.push_back({nums[i], nums[left], nums[right]});
left++, right--;
//对两个指针去重,前提是这个地方已经作为答案了,所以放在else中
while (left < right && nums[left - 1] == nums[left]) left++;
while (left < right && nums[right + 1] == nums[right]) right--;
}
}
//对i指针去重
i++;
while (i < n && nums[i - 1] == nums[i]) i++;
}
return ret;
}
};
4、力扣18. 四数之和
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> ret;
//1、排序
sort(nums.begin(), nums.end());
int n = nums.size();
//2、利用双指针思想求解
for (int i = 0; i < n;) //固定数a
{
//利用三数之和思路求解
for (int j = i + 1; j < n;) //固定数b
{
int left = j + 1, right = n - 1;
//数据范围为-10亿~10亿,那三个数相+-一定超int[-21亿~21亿]
//所以这里用long long才行,强转即可
long long t = (long long)target - nums[i] - nums[j];
while (left < right)
{
int sum = nums[left] + nums[right];
if (sum > t) right--;
else if (sum < t) left++;
else
{
ret.push_back({nums[i],nums[j],nums[left++],nums[right--]});
//对双指针去重
while (left < right && nums[left-1] == nums[left]) left++;
while (left < right && nums[right+1] == nums[right]) right--;
}
}
//对j指针指向的数去重
j++;
while (j < n && nums[j - 1] == nums[j]) j++;
}
//对i指针指向的数去重
i++;
while (i < n && nums[i - 1] == nums[i]) i++;
}
return ret;
}
};
5、四数相加 II
本题是四个独立的数组,只要找到A[i] + B[j] + C[k] + D[l] = 0就可以,不用考虑有重复的四个元素相加等于0的情况,即不用去重
解题步骤:
- 定义一个unordered_map,key放a和b两数之和,value 放a和b两数之和出现的次数。
- 遍历大A和大B数组,统计两个数组元素之和,和出现的次数,放到map中。
- 定义int变量count,用来统计 a+b+c+d = 0 出现的次数。
- 在遍历大C和大D数组,找到如果 0-(c+d) 在map中出现过的话,就用count把map中key对应的value也就是出现次数统计出来。
- 最后返回统计值 count 就可以了
C++代码:
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
//利用unordered_map的哈希表来实现快速查找。unordered_map中最多存储n^2个不同的key,
//因此空间复杂度为O(n^2)。
unordered_map<int, int> umap;
int count = 0;
for (int a : nums1){
for (int b : nums2){
umap[a + b]++;//出现次数++
}
}
for (int c : nums3){
for (int d : nums4){
if (umap.find(0 - (c + d)) != umap.end()){
count += umap[0 - (c + d)];
}
}
}
return count;
}
};
Java代码:
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
Map<Integer, Integer> map = new HashMap<>();
int count = 0;
for (int a : nums1){
for (int b : nums2){
int sum = a + b;
//getOrDefault的第一个参数是key,第二个参数的默认值,如果key存在则返回其出现次数,否则返回
//自己设置的默认值,比如这里默认值就是0,key不存在则返回0
map.put(sum, map.getOrDefault(sum, 0) + 1);
}
}
for (int c : nums3){
for (int d : nums4){
count += map.getOrDefault(0 - c - d, 0);
}
}
return count;
}
}