练习题 三数之和

题目

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组。

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。

示例 2：

输入：nums = [0,1,1]
输出：[]
解释：唯一可能的三元组和不为 0 。

示例 3：

输入：nums = [0,0,0]
输出：[[0,0,0]]
解释：唯一可能的三元组和为 0 。

提示：

• 3 <= nums.length <= 3000
• -105 <= nums[i] <= 105

提交代码

//三数之和

//双指针
//先判断（数1）和（数2、数3）之间的关系，再判断（数2）和（数3）的关系 
//判断三个数之间的关系：三数之和是否为0
//求所有的解
//求的解不能重复 
//数组无序 

#include<iostream>
#include<vector>
//包含sort()
#include<algorithm> 
using namespace std;

vector<int> nums;//整数数组 
int n;//数组元素个数
vector<vector<int>> result;//存储结果 

//双指针算法 
void ts(){
	int i = 0;//第一次双指针算法的指针之一：i指针指向数1，另外两个指针分别固定指向n - 2和i + 1 
	int p,q;//第二次双指针算法的两个指针
	int last = 100001;//存储上一次的i值
	
	//两次双指针算法
	
	//先判断数1和（数2，数3）的位置
	//第一次双指针算法 
	//确定数1可能的位置 
	while(i < n - 2){
		if(nums[i] == last){
			i++;
		}else if(nums[i] + nums[n - 2] + nums[n - 1] == 0){
			//找到一个答案
			vector<int> res;//暂时存储答案
			res.push_back(nums[i]); 
			res.push_back(nums[n - 2]);
			res.push_back(nums[n - 1]);
			result.push_back(res);
			//printf("%d %d %d\n",nums[i],nums[n - 2],nums[n - 1]);
			//找下一个答案
			last = nums[i];
			i++;
		}else if(nums[i] + nums[n - 2] + nums[n - 1] < 0){
			//数1在i位置的情况下不可能再有答案了 
			last = nums[i];
			i++; 
		}else if(nums[i] + nums[i + 1] + nums[i + 2] > 0){
			//不可能再有答案了 
			break;
		}else if(nums[i] + nums[n - 2] + nums[n - 1] > 0){
			//第二次双指针算法 
			p = i + 1;
			q = n - 1;
			int last2 = 100001;//存储上一次存储的p值，防止重复 
			
			while(p < q){
				if(nums[p] == last2){
					p++;
				}else if(nums[i] + nums[p] + nums[q] == 0){
					//找到一个答案
					vector<int> res;//暂时存储答案
					res.push_back(nums[i]); 
					res.push_back(nums[p]);
					res.push_back(nums[q]);
					result.push_back(res);
					//printf("%d %d %d\n",nums[i],nums[p],nums[q]);
					//找下一个答案
					last2 = nums[p]; 
					p++;
					q--; 
				}else if(nums[i] + nums[p] + nums[q] < 0){
					last2 = nums[p];
					p++;
				}else if(nums[i] + nums[p] + nums[q] > 0){
					q--;
				}
			}
			//注意修改条件，防止死循环 
			last = nums[i];
			i++;
		}
	} 
}

int main(){
	//输入数组 
	int t;//存储整数 
	
	while(cin.peek() != '\n'){
		scanf("%d",&t);
		nums.push_back(t);
	}
	
	//-------------------------------
	
	n = nums.size();//数组长度 
	
	//给数组排序
	sort(nums.begin(),nums.end()); 
	
	//双指针算法 
	ts(); 
	
	//输出结果
	for(int i = 0;i < result.size();i++){
		for(int j = 0;j < 3;j++){
			printf("%d ",result[i][j]);
		}
		printf("\n");
	} 
	
	return 0;
} 

总结

解题思路：求解三数之和，可以类比两数之和，可以使用暴力解法判断三个数的关系，可以使用“双指针”优化时间。需要使用多次双指针，本题一共需要两次。将数组排序后，第一次双指针判断最小的数可能是哪个，直接排除不可能的最小的数；第二次双指针判断第二大和最大的数可能是哪个。共需要三个指针下标，第一次双指针判断时，如果最小数加上最大两数小于0，则右移最小数指针，若大于0，则再进行第二次双指针，判断区间就是[i+1,n-1]，若最小数加上紧挨着的右边的两个数大于0，则不可能还有答案了，可以直接跳出循环，输出结果。

注意：题目要求不能有重复解，所以用last和last2分别记录最小数和次小数，若最小数和last相等或次小数和last2相等则不再判断，用于排除重复解。