码迷,mamicode.com
首页 > 其他好文 > 详细

【leetcode】3Sum

时间:2014-12-28 20:43:43      阅读:117      评论:0      收藏:0      [点我收藏+]

标签:

3Sum

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

Note:

  • Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
  • 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)
 
先对数组进行排序,从头开始扫描,扫描的数作为固定的数。
a=num[i];
 
当固定了一个数以后,
选取b=num[i+1]=num[j]
选取c=num[n-1]=num[k]
 
 
若a+b+c=0
则找到了一个a,b,c,记录下来,j++,k--继续扫描
若a+b+c<0
则j++
若a+b+c>0
则k--
 
 
要注意去重的问题:
1.若num[i]=num[i-1],则跳过
2.若num[j]=num[j-1],则跳过
3.若num[k]=num[k+1],则跳过
 
 
 1 class Solution {
 2 public:
 3     vector<vector<int> > threeSum(vector<int> &num) {
 4        
 5         int n=num.size();
 6         sort(num.begin(),num.end());
 7        
 8         int i,j,k;
 9         int a,b,c;
10         vector<vector<int> > result;
11        
12         for(i=0;i<n-2;i++)
13         {
14             j=i+1;
15             k=n-1;
16            
17             while(j<k)
18             {
19                 a=num[i];
20                 if(i>0&&num[i]==num[i-1])
21                 {
22                     break;
23                 }
24                 b=num[j];
25                 if(j>i+1&&num[j]==num[j-1])
26                 {
27                     j++;
28                     continue;
29                 }
30                
31                 c=num[k];
32                 if(k<n-1&&num[k]==num[k+1])
33                 {
34                     k--;
35                     continue;
36                 }
37                
38                 if(a+b+c==0)
39                 {
40                     vector<int> tmp(3);
41                     tmp[0]=a;
42                     tmp[1]=b;
43                     tmp[2]=c;
44                     result.push_back(tmp);
45                     j++;
46                     k--;
47                 }
48                 else if(a+b+c<0)
49                 {
50                     j++;
51                 }
52                 else if(a+b+c>0)
53                 {
54                     k--;
55                 }
56             }
57         }
58        
59         return result;
60        
61     }
62 };
63  

 

【leetcode】3Sum

标签:

原文地址:http://www.cnblogs.com/reachteam/p/4190319.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!