LeetCode刷题-三数之和

三数之和

题目

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

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

样例

1. 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
   满足要求的三元组集合为：
   [
   [-1, 0, 1],
   [-1, -1, 2]
   ]

Code

vector<vector<int>> threeSum(vector<int>& nums) {
	vector<vector<int>> result;
	if(nums.size()<3) {
		return result;
	}
	sort(nums.begin(), nums.end());
	const int target=0;

	auto last=nums.end();
	for(auto i=nums.begin(); i<last-2; ++i) {
		auto j=i+1;
		if(i>nums.begin() && *i==*(i-1)) {
			continue;
		}
		auto k=last-1;
		while(j<k) {
			if(*i + *j + *k < target) {
				++j;
				while(*j == *(j-1) && j<k) {
					++j;
				}
			} else if(*i + *j +*k > target) {
				--k;
				while(*k == *(k+1) && j<k) {
					--k;
				}
			} else {
				result.push_back({*i,*j,*k});
				++j;
				--k;
				while(*j == *(j-1) && *k==*(k+1) && j<k) {
					++j;
				}
			}
		}
	}
	return result;
}