标签:
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] ]
思路:
首先说点题外话,对于2Sum,3Sum,以及后面的4Sum,kSum等是一个系列的,具体的讨论在这篇文章中有提到,大家感兴趣的可以看看:http://www.sigmainfy.com/blog/summary-of-ksum-problems.html
回到正文,这道题很容易想到的是暴力解法,这也是我们通常采用的办法,但是这样的方法的复杂度是O(n3),再加上去重的复杂度,这样的代码在leetcode是无法成功运行的。
对于简化的思路,我自己没有想出来,网上参考的是:【LeetCode】3Sum 解题报告
我们不妨先对数组排个序。排序之后,我们就可以对数组用两个指针分别从前后两端向中间扫描了,如果是 2Sum,我们找到两个指针之和为target就OK了,那 3Sum 类似,我们可以先固定一个数,然后找另外两个数之和为第一个数的相反数就可以了。代码不难,先看了再说。
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 }
标签:
原文地址:http://www.cnblogs.com/mukekeheart/p/5719641.html