LeetCode(M)threesum

来源:互联网 发布:oppe软件商店 编辑:程序博客网 时间:2024/05/15 10:22

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]
]

class Solution {public:    vector<vector<int>> threeSum(vector<int>& nums) {        if(nums.size()<3) return{};        vector<vector<int>> v;        sort(nums.begin(),nums.end());        for(int i=0;i<nums.size();++i){            int target=-nums[i];            if(target<0) break  ;                int j=i+1,k=nums.size()-1;                while(j<k){                    if(nums[j]+nums[k]==target){                       v.push_back({nums[i],nums[j],nums[k]});                    //   while(j+1<k&&nums[j]==nums[j+1]) j++;while(j<k-1&&nums[k]==nums[k-1]) k--;//point1             int v1=nums[j],v2=nums[k]; while(j<k&&nums[j]==v1) j++;while(j<k&&nums[k]==v2) k--;//point2                    }                    if(nums[j]+nums[k]<target) ++j;                    if(nums[j]+nums[k]>target)  --k;            }            while(i+1<nums.size()&&nums[i]==nums[i+1]) i++;//pointA                    }                    return v;    }};

在LintCode上写过这题,http://blog.csdn.net/jang1996/article/details/53780446
这次,在leetCode上超时了,加入point2、pointA两句才过。最开始我将point2写成point1,犯错了。
point1的意思是前一个数与后一个数相等,那么++或–;point2的意思是,在后面的数里,凡是和nums[j]或nums[k]相等的,都跳过,以免在v里重复。

0 0
原创粉丝点击