[leetcode]15. 3Sum

Description

Given an array nums of n integers, are there elements a, b, c in nums 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.

Example:

Given array nums = [-1, 0, 1, 2, -1, -4],

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

Analysis

思路与Sum2类似，不过本题是三个数之和为0。该方法可只完整遍历数组一次就找出所有解，但时间复杂度仍为n^n。
首先将nums数组中所有数字升序排序。
在遍历第一遍数组的时候（事实上一共也只遍历了一遍），记下当前的数字nums[i]（作为三个数字中的第一个数字），若想得到一个三数和为0的解，其余两数之和应为target = -nums[i]。
令第二数为front，从i+1开始；第三数为back，从最后一个数nums[i].size()-1开始。在遍历第一个数字i时，在保持front永远小于back的情况下，计算front和back数字之和，若小于target，则将back向前移一个值；若大于target，则将front向后移一个值；若等于target，则表示发现了一个解，将这个三元组的解加入要返回的vector res中。
由于题目要求不能有重复解，则在更新front/back/i的值时，判断新值是否与原来解中对应数字的值相等。若相等，则继续++或- -，直到遇到不同的数字。

Code

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        if(nums.size()<3)
            return res;
        
        std::sort(nums.begin(), nums.end());
        
        for(int i=0; i<nums.size(); i++){
            int target = -1 * nums[i];
            if(target<0)    break;
            
            int front = i+1;
            int back = nums.size()-1;
            
            while(front<back){
                int sum = nums[front] + nums[back];
                if(sum<target)
                    front++;
                else if(sum>target)
                    back--;
                else{
                    vector<int> triplet(3,0);
                    triplet[0] = nums[i];
                    triplet[1] = nums[front];
                    triplet[2] = nums[back];
                    res.push_back(triplet);
                    while(front<back && nums[front]==triplet[1])
                        front++;
                    while(front<back && nums[back]==triplet[2])
                        back--;
                }
            }
            
            while(i+1<nums.size() && nums[i+1]==nums[i]){
                i++;
            }
        }
        
        return res;
    }

};