LeetCode-15. 三数之和

题目描述

原题链接

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组。

问题分析

先对整体数组进行升序排序，并指定三个指针i、j、k。

指针k从下标 0 开始遍历数组，剩下的两个指针i和j从指针k的右区间两端向内侧分别进行滑动。这里记指针i向右滑动，指针j向左滑动。

• 若nums[i]+nums[j]+nums[k] == 0：则将该三元组加入结果集。为避免出现重复解，要判断i和j的下个元素是否重复，若重复则跳过。
• 若nums[i]+nums[j]+nums[k] < 0：左指针i向右移动。
• 若nums[i]+nums[j]+nums[k] > 0：右指针j向左移动。

同时，指针k在遍历过程中也需要跳过重复元素，避免出现重复解。

程序代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
      vector<vector<int>> res;
      sort(nums.begin(), nums.end());
      int n = nums.size();
      for(int k = 0; k < n-2; k++) {
        if(nums[k] > 0)  break;
        if(k > 0 && nums[k] == nums[k-1])  continue;
        int i = k + 1, j = n - 1;
        while(i < j) {
          if(nums[k] + nums[i] + nums[j] < 0) {
            i++;
          }
          else if(nums[k] + nums[i] + nums[j] == 0) {
            vector<int> t = {nums[k], nums[i], nums[j]};
            res.push_back(t);
            while(i < j && nums[i] == nums[i+1]) {
              i++;
            }
            while(i < j && nums[j] == nums[j-1]) {
              j--;
            }
            i++;
            j--;
          }
          else {
            j--;
          }
        }
      }
      return res;
    }
};