码迷,mamicode.com
首页 > 编程语言 > 详细

[LeetCode] 15. 3Sum Java

时间:2018-02-27 12:41:49      阅读:168      评论:0      收藏:0      [点我收藏+]

标签:元素   java   ++   let   font   枚举   note   nts   ret   

题目: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]
]

题意及分析:

求3Sum的思路如下:

  • 先整体排一次序
  • 然后遍历数组,固定一个元素,对其后的元素求2Sum。

比如排好序以后的数字是 a b c d e f, 那么第一次固定a, 在剩下的 b c d e f 中进行2sum, 完了以后第二次枚举b, 只需要在 c d e f 中进行2sum,这样就避免了重复。

public class Solution {
    public List<List<Integer>> threeSum(int[] nums) {   
        List<List<Integer>> res=new ArrayList<>();
        if (nums == null || nums.length < 3) return res;  
        Arrays.sort(nums);
        int n=nums.length;
        
        for(int i=0;i<n-2;i++){
            if (i > 0 && nums[i] == nums[i-1]) continue;
            int a=-nums[i];
            int left=i+1;
            int right=n-1;
            while(left<right){
                if((nums[left]+nums[right])>a){
                    right--;
                }else if((nums[left]+nums[right])<a){
                    left++;
                }else{
                    List<Integer> tmp=new ArrayList<Integer>();
                    tmp.add(nums[i]);
                    tmp.add(nums[left]);
                    tmp.add(nums[right]);
                    while (left < right && nums[left] == nums[left+1]) left++;  
                    while (left < right && nums[right] == nums[right-1]) right--;  
                    left++;
                    right--;
                    res.add(tmp);
                };
            }
        }
        return res;
    }
}

 

[LeetCode] 15. 3Sum Java

标签:元素   java   ++   let   font   枚举   note   nts   ret   

原文地址:https://www.cnblogs.com/271934Liao/p/8478169.html

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