算法——四数之和
leetcode18题,经典题目。
题意:
给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。
这一类题目包括两数之和、三数之和、四数之和、还可以拓展到K数之和。解题时可以从两数之和以及三数之和的求解中拓展开。
解体思路:
- 排序+双指针(适当剪枝可提高运行速度)
两数之和:
对数组nums进行升序排序,然后用两个指针i和j分别指向数组的头和尾,指针i和j向彼此靠拢,并判断nums[i]+nums[j]是否等于target。
三数之和:
在两数的基础上多了一个外层循环,在最外层循环中固定一个数(坐标为m),在内层循环中采用双指针指向另外两个数,判断nums[m]+nums[i]+nums[j]是否等于target。
四数之和:
在三数的基础上又多加一个外层循环,两个外层循环可固定两个数,内层双指针指向剩下的两个数。
//两数之和
vector<vector<int>> twoSum(vector<int>& nums, int target) {
vector<vector<int>> res;
if(nums.size()<2) return res;
sort(nums.begin(),nums.end());
int i=0,j=nums.size()-1;
while(i<j){
int sum=nums[i]+nums[j];
if(sum==target){
res.emplace_back(vector<int>{nums[i],nums[j]});
while(i<j&&nums[i]==nums[i+1]) i++;//去除重复元素
while(i<j&&nums[j]==nums[j-1]) j--;
i++;
j--;
}
else if(sum<target) i++;
else j--;
}
return res;
}
//三数之和
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
if(nums.size()<3) return res;
sor