题号 15
题目描述
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
注意:以下三种解题思路,只有第三种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