leetcode 15.三數之和
給定一個包含 n 個整數的陣列 nums
,判斷 nums
中是否存在三個元素 a,b,c ,使得 a + b + c = 0 ?找出所有滿足條件且不重複的三元組。
注意:答案中不可以包含重複的三元組。
例如, 給定陣列 nums = [-1, 0, 1, 2, -1, -4], 滿足要求的三元組集合為: [ [-1, 0, 1], [-1, -1, 2] ]
class Solution { public: vector<vector<int>> threeSum(vector<int>& nums) { vector<vector<int>> res; sort(nums.begin(),nums.end()); //對於陣列進行升序排序 if(!nums.empty() && nums.back() < 0) return {}; //如果陣列nums不為空且陣列的最後一項小於0,直接返回空 for(int i = 0; i < nums.size(); i++){ if(nums[i] > 0) break; //如果陣列開始第一個數字大於0,那麼後面的數字都是大於0,直接迴圈停止 if(i > 0 && nums[i] == nums[i-1]) continue; //如果陣列前一個和後一個相等的話,直接跳過後一個,防止重複 int target = 0 - nums[i]; int m = i + 1; //左指標為m int n = nums.size() - 1; //右指標為n while(m < n){ if(nums[m] + nums[n] == target){ res.push_back({nums[i],nums[m],nums[n]}); //如果nums[m]+nums[n]=target,直接記錄三個數字的值 while(m < n && nums[m] == nums[m+1]) m++; //如果m<n且後一個和前一個數字相等,那麼m++ while(m < n && nums[n] == nums[n-1]) n--; //如果m<n且後一個和前一個數字相等,那麼n-- m++; //如果以上條件不滿足,直接將m++,n-- n--; } else if(nums[m] + nums[n] < target) m++; //如果兩數相加之和小於target,則將左指標m++ else n--; //如果兩數相加之和大於target,則將右指標n-- } } return res; } };