LeetCode 15. 三数之和

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>>ans;
        sort(nums.begin(),nums.end());
        for(int i = 0;i < nums.size();i++){
            if(i != 0 && nums[i] == nums[i - 1]) continue;
            for(int j = i + 1;j < nums.size();j++){
                if(j != i + 1 && nums[j] == nums[j - 1])continue;
                int start = j + 1,end = nums.size()-1,mid = (start + end)/2;
                int target = -(nums[i] + nums[j]);
                while(start <= end){
                    if(nums[mid] > target) end = mid - 1;
                    else if(nums[mid] < target) start = mid + 1;
                    else break;
                    mid = (start + end)/2;
                }
                if(start <= end){
                    vector<int>temp;
                    temp.push_back(nums[i]);
                    temp.push_back(nums[j]);
                    temp.push_back(nums[mid]);
                    ans.push_back(temp);
                }
            }
        }
        return ans;
    }
};

解题思路：这个题目暴力好像会超时，所以在找最后一个元素的时候用折半查找，这样的话整个程序就优化到了O(n^2logn)，这个题目还有一个需要处理的地方的就是需要处理重复的结果，因为一开始对该序列进行排序，所以在进行遍历的时候只要判断一下这个数和上一个数是否相等即可，相等的话直接跳过就好了。