前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >No.015 3Sum

No.015 3Sum

作者头像
mukekeheart
发布2018-02-27 15:34:10
6270
发布2018-02-27 15:34:10
举报
文章被收录于专栏:mukekeheart的iOS之旅

15. 3Sum

  • Total Accepted: 131800
  • Total Submissions: 675028
  • Difficulty: Medium

  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: 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] ] 思路:   首先说点题外话,对于2Sum,3Sum,以及后面的4Sum,kSum等是一个系列的,具体的讨论在这篇文章中有提到,大家感兴趣的可以看看:http://www.sigmainfy.com/blog/summary-of-ksum-problems.html

  回到正文,这道题很容易想到的是暴力解法,这也是我们通常采用的办法,但是这样的方法的复杂度是O(n3),再加上去重的复杂度,这样的代码在leetcode是无法成功运行的。   对于简化的思路,我自己没有想出来,网上参考的是:【LeetCode】3Sum 解题报告

  我们不妨先对数组排个序。排序之后,我们就可以对数组用两个指针分别从前后两端向中间扫描了,如果是 2Sum,我们找到两个指针之和为target就OK了,那 3Sum 类似,我们可以先固定一个数,然后找另外两个数之和为第一个数的相反数就可以了。代码不难,先看了再说。

代码语言:javascript
复制
  1 public class No_015 {
  2     /*
  3      * 方法一:暴力解法
  4      */
  5     public List<List<Integer>> threeSum(int[] nums) {
  6         List<List<Integer>> res = new ArrayList<List<Integer>>() ;
  7         if(nums == null || nums.length < 3){
  8             return res ;
  9         }
 10         List<Integer> list = null ;
 11         for(int i = 0 ; i < nums.length-2 ; i++){
 12             for(int j = i+1 ; j < nums.length-1 ; j++){
 13                 for(int k = j+1 ; k < nums.length ; k++){
 14                     if((nums[i] + nums[j] + nums[k]) == 0){
 15                         list = toRange(nums,i,j,k) ;
 16                         if(!isContain(res,list)){
 17                             res.add(list) ;
 18                         }  
 19                     }
 20                 }
 21             }
 22         }
 23         return res ; 
 24     }    
 25     
 26     /*
 27      * 对三个数之和为0的数加入list的时候进行排序,方便后面去重
 28      * 
 29      * 唉,怎么没想到一开始就对曾格格数组进行排序呢???
 30      */
 31     private List<Integer> toRange(int [] nums ,int i , int j , int k){
 32         List<Integer> list = new ArrayList<Integer>() ;
 33         int max = nums[i]>nums[j]?nums[i]:nums[j] ;
 34         int min = nums[i]+nums[j]-max ;
 35         max = max>nums[k]?max:nums[k] ;
 36         min = min<nums[k]?min:nums[k] ;
 37         list.add(min) ;
 38         list.add(nums[i]+nums[j]+nums[k]-max-min) ;
 39         list.add(max) ;
 40         return list ;
 41     }
 42     /*
 43      * 检查是否有重复
 44      */
 45     private boolean isContain(List<List<Integer>> res , List<Integer> list){
 46         if(res == null || res.size() == 0){
 47             return false ;
 48         }
 49         for(List<Integer> each:res){
 50             if(each.get(0) == list.get(0) && each.get(1) == list.get(1)){
 51                 return true ;
 52             }
 53         }
 54         return false ;
 55     }
 56     
 57     /*
 58      * 方法二:首先进行排序,然后根据排序之后的数组,从两边到中间的办法查询,这样复杂度为o(n^2)
 59      */
 60     List<List<Integer>> ret = new ArrayList<List<Integer>>();
 61     public List<List<Integer>> threeSum2(int[] num) {
 62 
 63         if (num == null || num.length < 3)
 64             return ret;
 65 
 66         Arrays.sort(num);
 67 
 68         int len = num.length;
 69         for (int i = 0; i < len - 2; i++) {
 70             if (i > 0 && num[i] == num[i - 1])
 71                 continue;
 72             find(num, i + 1, len - 1, num[i]); // 寻找两个数与num[i]的和为0
 73         }
 74 
 75         return ret;
 76     }
 77 
 78     public void find(int[] num, int begin, int end, int target) {
 79         int l = begin, r = end;
 80         while (l < r) {
 81             if (num[l] + num[r] + target == 0) {
 82                 List<Integer> ans = new ArrayList<Integer>();
 83                 ans.add(target);
 84                 ans.add(num[l]);
 85                 ans.add(num[r]);
 86                 ret.add(ans); // 放入结果集中
 87                 while (l < r && num[l] == num[l + 1])
 88                     l++;
 89                 while (l < r && num[r] == num[r - 1])
 90                     r--;
 91                 l++;
 92                 r--;
 93             } else if (num[l] + num[r] + target < 0) {
 94                 l++;
 95             } else {
 96                 r--;
 97             }
 98         }
 99     }  
100     
101 }
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2016-07-29 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 15. 3Sum
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档