leetcode #15 三数之和 | 刷题之路第一站——数组类相关问题

题号 15

题目描述

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

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

示例:

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

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

链接：leetcode #15 三数之和

注意：以下三种解题思路，只有第三种AC了，可以直接查看第三种。

解题思路1【暴力法】：

最直观的想法就是采用三层for循环，依次遍历所有的三数之和，将符合条件的三元组存放于结果集合中，注意不要存入重复的三元组。
时间复杂度分析： 时间复杂度为O(n^3)
注意： 该方法超出时间限制，未AC。

代码实现：

class Solution {
   
public:
	vector<vector<int>> threeSum(vector<int>& nums) {
   
		sort(nums.begin(), nums.end());    //将给定数组nums进行排序
		vector<vector<int>> ans;    //ans是存放和为0、且不重复的三元组的集合
		vector<int> med;
		for (int i = 0; i < nums.size(); i++)    //三重for循环
			for (int j = i + 1; j < nums.size(); j++)
				for (int k = j + 1; k < nums.size(); k++)
				{
   
					if (nums[i] + nums[j] + nums[k] == 0)    //如果三者和为0，并且结果集合中不包含三者的任意组合，则将该三元组存入结果结合ans中
					{
   
						med.push_back(nums[i]);
						med.push_back(nums[j]);
						med.push_back(nums[k]);
						if (!count(ans.begin(), ans.end(), med))    //判断结果集合中是否已经存在该三元组
						{
   
							ans.push_back(med