1. 程式人生 > >leetcode 15.三數之和

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;            } };