LeetCode 15.三数之和

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

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

示例：

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

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

tips: 排序+双指针，时间复杂度O(n^2)

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
		
		vector<vector<int>> res;
		if(nums.size()<=0) {
			return res;
		} 
		sort(nums.begin(),nums.end());
		int target,pre_index,post_index,sum;
		for (int i = 0; i < nums.size()-1; i++)
		{
			target=0-nums[i];
			pre_index=i+1;
			post_index=nums.size()-1;
			
			while(pre_index<post_index) {
                sum=nums[pre_index]+nums[post_index];
				if(sum==target) {
					res.push_back({nums[i],nums[pre_index],nums[post_index]});
					while(i<nums.size()-1 && nums[i+1]==nums[i]) {
						i++;
					}
					pre_index++;
					post_index--;
					while(pre_index<nums.size()&&nums[pre_index-1]==nums[pre_index]) {
						pre_index++;
                    }
					while(post_index>0&&post_index+1<nums.size()&&nums[post_index+1]==nums[post_index]) {
						post_index--;
                    }
				} else if(sum<target) {
					pre_index++;
					while(pre_index<nums.size()&&nums[pre_index-1]==nums[pre_index]) {
						pre_index++;
                    }
				} else {
					post_index--;
					while(post_index>0&&post_index+1<nums.size()&&nums[post_index+1]==nums[post_index]) {
						post_index--;
                    }
				}
			}
		}
		return res;
    }
};