15. 三数之和
题目
给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请
你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
题目链接
. - 力扣(LeetCode)
画图 和 文字 分析
这道题和 两数之和等于一个值 大体思路是一样的,都是 排序 + 双指针思想
排完序后,我们定义三个指针,一个指向最后一个元素的位置,一个指向首元素的位置,另一个首元素的后一个位置
举例:
输入: [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
先固定 k不动
- 如果三者指向的值相加为 0 ,则记录数据 ,再 j++ , k--
- 如果三者指向的值 < 0 ,则 j++
- 如果三者指向的值 > 0 ,则 k--
当 i >= j (结束里层循环)
再 i++ , j = i + 1 , k = n - 1
直到 i + 1 >= k (外层循环)
做到以上,只能说完成了完成了不漏掉每一种情况,但现在还有去重的关键一步
去重需要我们在前面的基础上做更改:
第一种情况:
走完上面的步骤 :
判断现在 j 所指的内容 和 j - 1 所指内容是否相同,直到不相同为止(这里需要一个循环,此时要么,j 指向一个不和之前相重复的数,要么越界)
判断 k 同理
上面是里层循环的去重,外层循环也可以去重
当结束里层循环,完成后面的步骤 :
判断 i 所指向的内容 和 i - 1所指向的内容是否相同,直到不相同为止
注意:
去重的时候,因为循环的缘故,一定要防止越界
代码
class Solution { public:vector<vector<int>> threeSum(vector<int>& nums) {vector<vector<int>> t;sort(nums.begin(),nums.end());int i = 0;while(i + 2 <= nums.size() - 1){int j = i + 1;int k = nums.size() - 1;;while(j < k){if(nums[i] + nums[j] + nums[k] > 0){k--;}else if(nums[i] + nums[j] + nums[k] < 0){j++;}else{vector<int> x;x.push_back(nums[i]);x.push_back(nums[j]);x.push_back(nums[k]);t.push_back(x);int n = nums[j];int m = nums[k];k--;j++;while(j < k && n == nums[j]){j++;}while(j < k && m == nums[k]){k--;}}}int h = nums[i];i++;while(i + 2 <= nums.size() - 1 && h == nums[i]){i++;}}return t;} };
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
你可以按 任意顺序 返回答案 。
题目链接
. - 力扣(LeetCode)
画图 和 文字 分析
在 leetcode 15. 三数之和 基础之上做出的改变
思想:排序 + 双指针思想
定义四个指针,三个指针分别指向前三个元素的位置,第四个指针指向最后一个元素的位置
举例:
输入:nums = [1,0,-1,0,-2,2], target = 0
输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
后面的三个指针和之前的做法一模一样,第一个指针在里层所有循环结束后++
再判断现在 a 所指向的元素和 a - 1 所指向的元素是否相同
直到 a + 2 >= d (外层循环结束)
注意:
- 注意越界情况
- 判断四数之和是否得到一个值,这里容易由于数据过大发生整型溢出现象,可以改成 longlong 类型 或者针对处理这一可能
代码
class Solution { public:bool iscompare(int &a,int &b,int &c,int &d,int &target){if(target < 0 && (a >= 0 && b >= 0 && c >= 0 && d >= 0)){return false;}else if(target > 0 && (a <= 0 && b <= 0 && c <= 0 && d <= 0)){return false;}else if(target == 0 && ((a > 0 && b > 0 && c > 0 && d > 0) || (a > 0 && b > 0 && c > 0 && d > 0))){return false;}else{return true;}}vector<vector<int>> fourSum(vector<int>& nums, int target) {vector<vector<int>> s;int n = nums.size() - 1;sort(nums.begin(),nums.end());int a = 0;while(a + 2 < n){int b = a + 1;while(b + 1 < n){int c = b + 1;int d = n;while(c < d){if(!iscompare(nums[a],nums[b],nums[c],nums[d],target)){break;}if(nums[a] + nums[b] > target - nums[c] - nums[d] ){d--;}else if(nums[a] + nums[b] < target - nums[c] - nums[d] ){c++;}else{vector<int> t;t.push_back(nums[a]);t.push_back(nums[b]);t.push_back(nums[c]);t.push_back(nums[d]);s.push_back(t);int s1 = nums[c];int s2 = nums[d];d--;c++;while(c < d && nums[c] == s1){c++;}while(c < d && nums[d] == s1){d--;}}}int s3 = nums[b];b++;while(b + 1 < n && nums[b] == s3){b++;}}int s4 = nums[a];a++;while(a + 2 < n && nums[a] == s4){a++;}}return s;} };