【算法与数据结构】15、LeetCode三数之和

所有的LeetCode题解索引，可以看这篇文章——【算法和数据结构】LeetCode题解。

一、题目

二、双指针法

  思路分析：我们使用双指针法，但这道题因为要求数组三个元素的和，一共用到了三个指针，但在循环当中只有两个指针起作用，还是把解法称为双指针法。题目要求不能有重复的三元组，那么我们在程序中判断第i个元素是否和i-1哥元素相同，如果相同则跳过第i元素。while 语句当中，判断 nums[left] 和 nums[right] 的和是否大于target，即判断三元素和是否大于0，当跳出循环时，和必然小于等于0，进行判断，如果等于0添加进结果中，否则继续循环。
  程序如下：

class Solution {
public:
	// 双指针法
	vector<vector<int>> threeSum(vector<int>& nums) {
		int n = nums.size();
		sort(nums.begin(), nums.end());	// 升序排列
		vector<vector<int>> result;
		if (nums[0] > 0) return result;	// 最小值都大于零，不存在三元组和为0
		for (int i = 0; i < n; ++i) {
			if (i > 0 && nums[i] == nums[i - 1]) continue;	// i去重
			int right = n - 1;		// 最右端
			int target = -nums[i];	

			for (int left = i + 1; left < n; ++left) {
				if (left > i + 1 && nums[left] == nums[left - 1]) continue;	// left去重
				while (left < right && nums[left] + nums[right] > target) {
					--right;		
					// 三者之和大于0，right收缩，一直收缩到小于等于0或者left=right为止
				}
				if (left == right)	break;	// 不存在三元组，因此break。
				if (nums[left] + nums[right] == target) {
					result.push_back({ nums[i], nums[left], nums[right] });
					//result.push_back({ i, left, right });	// 调试用
				}
			}
		}
		return result;
	}
};

复杂度分析：

• 时间复杂度： $O(n^2)$。
• 空间复杂度：$O(1)$。

  简单修改一下输入参数和判断语句就可以输出和为target的三元组。这部分代码我们也会在四数之和当中用到。

class Solution {
public:
	// 双指针法 找到目标和为target的三元组
	vector<vector<int>> threeSum(vector<int>& nums, int target) {
		int n = nums.size();
		sort(nums.begin(), nums.end());	// 升序排列
		my_print1(nums, "排序后数组");
		vector<vector<int>> result;
		if (nums[0] > target) return result;	// 最小值都大于target，不存在三元组和为target
		for (int i = 0; i < n; ++i) {
			if (i > 0 && nums[i] == nums[i - 1]) continue;	// i去重
			int right = n - 1;		// 最右端
			int temp = -nums[i] + target;	

			for (int left = i + 1; left < n; ++left) {
				if (left > i + 1 && nums[left] == nums[left - 1]) continue;	// left去重
				while (left < right && nums[left] + nums[right] > temp) {
					--right;		
					// 三者之和大于0，right收缩，一直收缩到小于等于0或者left=right为止
				}
				if (left == right)	break;	// 不存在三元组，因此break。
				if (nums[left] + nums[right] == temp) {
					result.push_back({ nums[i], nums[left], nums[right] });
					//result.push_back({ i, left, right });	// 调试用
				}
			}
		}
		return result;
	}
};

三、完整代码

# include <iostream>
# include <string>
# include <vector>
# include <algorithm>
using namespace std;

void GeneratorVector(int arr[], int arr_len, vector<int>& v) {
	for (int i = 0; i < arr_len; i++) {
		v.push_back(arr[i]);
	}
}

void my_print1(vector<int>& v, string str) {
	cout << str << endl;
	for (vector<int>::iterator it = v.begin(); it < v.end(); ++it) {
		cout << *it << ' ';
	}
	cout << endl;
}

void my_print2(vector<vector<int>>& v, string str) {
	cout << str << endl;
	for (vector<vector<int>>::iterator vit = v.begin(); vit < v.end(); ++vit) {
		for (vector<int>::iterator it = (*vit).begin(); it < (*vit).end(); ++it) {
			cout << *it << ' ';
		}
		cout << endl;
	}
}

class Solution {
public:
	// 双指针法
	vector<vector<int>> threeSum(vector<int>& nums) {
		int n = nums.size();
		sort(nums.begin(), nums.end());	// 升序排列
		vector<vector<int>> result;
		if (nums[0] > 0) return result;	// 最小值都大于零，不存在三元组和为0
		for (int i = 0; i < n; ++i) {
			if (i > 0 && nums[i] == nums[i - 1]) continue;	// i去重
			int right = n - 1;		// 最右端
			int target = -nums[i];	

			for (int left = i + 1; left < n; ++left) {
				if (left > i + 1 && nums[left] == nums[left - 1]) continue;	// left去重
				while (left < right && nums[left] + nums[right] > target) {
					--right;		
					// 三者之和大于0，right收缩，一直收缩到小于等于0或者left=right为止
				}
				if (left == right)	break;	// 不存在三元组，因此break。
				if (nums[left] + nums[right] == target) {
					result.push_back({ nums[i], nums[left], nums[right] });
					//result.push_back({ i, left, right });	// 调试用
				}
			}
		}
		return result;
	}
};

int main()
{
	int arr[] = {-1, 0, 1, 2, -1, -4};
	//int arr[] = { 0, 0, 0, 0 };
	//int arr[] = { -2,0,1,1,2 };
	int arr_len = sizeof(arr) / sizeof(int);
	vector<int> nums;
	GeneratorVector(arr, arr_len, nums);
	my_print1(nums, "目标数组：");
	Solution s1;
	vector<vector<int>> result= s1.threeSum(nums);	
	my_print2(result, "结果：");	
	system("pause");
	return 0;
}

end