15.三數之和

題目連結：15.三數之和(中等)

題目描述

給你一個包含 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
-10^5 <= nums[i] <= 10^5

題解

思路：

接下來如何移動 l 和 r 呢， 如果nums[i] + nums[l] + nums[r] > 0 就說明此時三數之和大了，因為陣列是排序後了，所以 r 下標就應該向左移動，這樣才能讓三數之和小一些。

如果 nums[i] + nums[l] + nums[r] < 0 說明 此時三數之和小了，l 就向右移動，才能讓三數之和大一些，直到 l 與 r 相遇為止。

程式碼(C++版本)

vector<vector<int>> threeSum(vector<int>& nums) {
  vector<vector<int>> result;
  sort(nums.begin(), nums.end());//排序，從小到大
​
  for (int i = 0; i < nums.size(); i++) {
    if (nums[i] > 0) {
      return result;
     }
    //第一個指標指的數不等於前面的數；資料是有序的，所以只需判斷當前的數和前一位數即可
    if (i > 0 && nums[i] == nums[i - 1]) {
      continue;
     }
    //左指標指向第一個指標所指的後一位數，右指標指向最後一位數
    for (int l = i + 1, r = nums.size() - 1; l < r;) {
      if (nums[i] + nums[l] + nums[r] < 0) {
        l++;
       }
      else if (nums[i] + nums[l] + nums[r] > 0) {
        r--;
       }
      else if (nums[i] + nums[l] + nums[r] == 0) {
        result.push_back({nums[i], nums[l], nums[r]});
        while (r > l && nums[r] == nums[r - 1]) r--;
        while (r > l && nums[l] == nums[l + 1]) l++;
        r--;
        l++;
       }
     }
   }
  return result;
}
 

分析：時間複雜度：O(N^2)

參考連結：

官方題解(三數之和)

程式碼隨想錄(三數之和)