題目如下:
給你一個包含 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),
轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/260172.html
標籤:其他
