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>> results;
        int len = nums.size();
        if (len < 3) {
            return results;
        }
        sort(nums.begin(), nums.end());
        for (int i = 0; i < len - 2; ++i) {
            while  ((i != 0) && (i < len - 1) && (nums[i] == nums[i - 1])) {
                ++i;
            }
            int start = i + 1, end = len - 1;
            while (start < end) {
                int sum = nums[i] + nums[start] + nums[end];
                if (sum == 0) {
                    results.push_back({ nums[i], nums[start], nums[end]});
                    ++start;
                    while ((start <= end) && (nums[start] == nums[start - 1])) {
                        ++start;
                    }
                    --end;
                    while ((start <= end) && (nums[end] == nums[end + 1])) {
                        --end;
                    }
                } else if (sum < 0) {
                    ++start;
                    while ((start <= end) && (nums[start] == nums[start - 1])) {
                        ++start;
                    }
                } else {
                    --end;
                    while ((start <= end) && (nums[end] == nums[end + 1])) {
                        --end;
                    }
                }
            }
        }
        return results;
    }
};

我的成绩

执行结果：通过
执行用时 : 128 ms, 在所有 C++ 提交中击败了35.26%的用户
内存消耗 : 20 MB, 在所有 C++ 提交中击败了7.60%的用户

一些想法

本道题我的想法是先将数组排好序，然后从数组的左端开始固定，对于剩下的内容使用双指针法进行判断，如果三数之和大于0，则将右端点向左移动，小于0则将左端点向右移动，等于0，则将左端点，右端点同时移动，需要注意的是移动的端点要得与移动之前内容不同。

执行用时为 28 ms 的范例

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        vector<vector<int>> ans;
        int len = nums.size();
        for(int i = 0; i < len; ++i){
            if(i > 0 && nums[i] == nums[i - 1])continue;

            int target = nums[i];
            if(target > 0){
                break;
            }
            int l = i + 1, r = len - 1;
            while(l < r){
                if(target + nums[l] + nums[r] < 0){
                    ++l;
                }else if(target + nums[l] + nums[r] > 0){
                    --r;
                }else{
                    ans.push_back({target, nums[l], nums[r]});
                    ++l;
                    --r;
                    while(l < r && nums[l] == nums[l - 1])++l;
                    while(l < r && nums[r] == nums[r + 1])--r;
                }
            }
        }
        return ans;
    }
};

思考

方法一样。