【算法day14】三数之和-CSDN博客

本文链接：https://blog.csdn.net/qinglingLS/article/details/146357547

三数之和

https://leetcode.cn/problems/3sum/description/

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。
注意：答案中不可以包含重复的三元组。

其实仔细想想，三个指针里面i是单向移动的？？还可以简化
在这里插入图片描述

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int nums_len = nums.size();
        sort(nums.begin(), nums.end());
        vector<vector<int>> ans;
        for (int left = 0; left < nums_len; left++) {
            if (left > 0 && nums[left] == nums[left - 1]) {
                continue;
            }
            int right = nums_len - 1;
            for (int mid = left + 1; mid < nums_len; mid++) {
                if (mid > left + 1 && nums[mid] == nums[mid - 1]) {
                    continue;
                }
                while (mid < right &&
                       nums[left] + nums[mid] + nums[right] > 0) {
                    right--;
                }
                if (mid >= right) {
                    break;
                }
                if (nums[left] + nums[mid] + nums[right] == 0) {
                    ans.push_back({nums[left], nums[mid], nums[right]});
                }
            }
        }
        return ans;
    }
};

在这里插入图片描述

vector<vector<int>> threeSum(vector<int>& nums) {
        int nums_len = nums.size();
        sort(nums.begin(), nums.end(), [](int a, int b) { return a < b; });
        vector<vector<int>> ans;
        for (int left = 0; left < nums_len; left++) {
            if (left > 0 && nums[left] == nums[left - 1]) {
                // 遇到重复的进行跳过
                continue;
            }
            int right = nums_len - 1;
            for (int mid = left + 1; mid < nums_len; mid++) {
                if (mid > left + 1 && nums[mid] == nums[mid - 1]) {
                    // 遇到重复的进行跳过
                    continue;
                }
                while (mid < right &&
                       nums[mid] + nums[left] + nums[right] > 0) {
                    right--;
                }
                if (right == mid) {
                    break;
                }
                if (nums[mid] + nums[left] + nums[right] == 0) {
                    ans.push_back({nums[left], nums[mid], nums[right]});
                }
            }
        }
        return ans;
    }