Leetcode 15. 3Sum

15. 3Sum

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note: The solution set must not contain duplicate triplets.

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

 

题目链接：https://leetcode.com/problems/3sum/description/

题目大意：给定数组S，求和为 0 的三个数的组合，要求组合不重复

解题思路：三重循环时间复杂度是O(n^3)，先快速排序，时数组递增有序。

                  第一层：第一个数按顺序遍历O(n) ；

                  第二层：第二个数从前往中间遍历，如果总数<0,下标增加，第三个数从最后往中间遍历，如果总数<0,下标增加，O(n) ；                     合起来时间复杂度为O(n^2)

代码：

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