【LeetCode Hot 100 三数之和】
题目如下:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
初始思路:
解题时的想法如下,这里是三数之和,所以会有三种变量,此时我的想法是先固定住一个变量,对于其他的两个变量可以使用双指针的办法来做。
代码如下:
vector<vector<int>> threeSum(vector<int>& nums) { vector<vector<int>> vecValue;
/*长度小于3,直接返回空*/ if(nums.size()<3) { return vecValue; }
/*先进行排序*/ sort(nums.begin(),nums.end()); for(int i=0;i<=nums.size()-3;i++) {
/*防止将相同的序列重复插入*/ if(i>0&&nums[i]==nums[i-1]) { continue; } int j=i+1,k=nums.size()-1,intTemp=0; while(j<k) {
/*同理,也是为了防止将相同的序列重复插入*/ if(j>i+1&&nums[j]==nums[j-1]) { j++; continue; } intTemp=nums[j]+nums[k]+nums[i];
/*值小于0,则将j++,增大三数之和*/ if(intTemp<0) { j++; }
/*值大于0,则将k--,减少三数之和*/ else if(intTemp>0) { k--; } else { vector<int> vecTemp; vecTemp.push_back(nums[i]); vecTemp.push_back(nums[j]); vecTemp.push_back(nums[k]); vecValue.push_back(vecTemp);
/*j自增,同时防止死循环*/ j++; } intTemp=0; } } return vecValue; }
复杂度:
时间复杂度:O(n*n)。
空间复杂度:忽略存储答案所占的空间,排序的空间复杂度即为本解法的空间复杂度,即log(N)。