LeetCode 15. 三数之和

1. 题目信息

链接：https://leetcode-cn.com/problems/3sum

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

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

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

2. 解题思路

暴力查找，时间复杂度O(n³)，会超时

1. 题目不要求输出下标，先从小到大排序
2. 排除特殊情况（元素少于3，全部大于0，全部小于0）
3. i从0到n-2；第一个数取nums[i]，第二个数left取nums[i+1]，第三个数right取nums[n-1]，判断三数之和sum
4. sum < 0, left下标+1；sum > 0, right下标-1；sum = 0，存入vector
5. 注意遇到相同的元素将下标直接移动

时间复杂度O(n²)

3. 代码

class Solution
{
public:
    vector<vector<int>> threeSum(vector<int>& nums) 
    {
    	vector<vector<int>> ans;
    	sort(nums.begin(), nums.end());//排序
    	int n = nums.size();
    	if(n < 3||nums.front()>0||nums.back()<0)
    		return ans;
    	int left, right, templeft, tempright, sum;
    	for(int i = 0; i < n-2; ++i)
    	{
    		if(nums[i] > 0)//全为正，找不到解了
    			break;
    		if(i > 0 && nums[i-1] == nums[i])//相同元素，跳过
                continue;
    		left = i+1, right = n-1;
    		while(left < right)
    		{
    			sum = nums[i]+nums[left]+nums[right];
    			if(sum > 0)
    				right--;
    			else if(sum < 0)
    				left++;
    			else
    			{
    				ans.push_back(vector<int> {nums[i], nums[left], nums[right]});
    				templeft = nums[left];
    				tempright = nums[right];
    				while(left < right && templeft == nums[++left])
    				{
                        //跳过相同元素
    				}
    				while(left < right && tempright == nums[--right])
    				{
                        //跳过相同元素
    				}
    			}
    		}
    	}
    	return ans;
    }
};