LeetCode-15. 3Sum

一、问题描述

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.

二、解题思路

求解int数组中满足三数之和为0的组合。首先判断数组的长度是否大于2，如果不满足则返回空的列表；然后将数组中的元素按照从小到大的顺序排序，判断数组中的首元素和尾元素的乘积，如果积大于等于零，则返回空列表（这里有一个特殊情况就是三个元素都为0的情况），反之则开始寻找数组中满足条件的组合。定义三个位置变量，i（范围是从0到数nums.length-3）,j（范围是 从i+1到k）,k（范围是从nuns.length-1到i），固定变量I后，计算nums[i]+nums[j]+nums[k]的和，如果等于零，则存入列表中；如果大于零，则k--;如果小于零，则j++；

三、代码

public class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
              List<List<Integer>> result=new ArrayList<List<Integer>>();
        if(nums==null || nums.length<=2)
            return result;
        Arrays.sort(nums);
        if(nums[0]*nums[nums.length-1]>=0 && nums[0]!=nums[nums.length-1])
            return result;
        //int middle=findMiddle(0,nums.length-1,nums);
       // if(middle+1<nums.length-middle-1){
            for(int i=0;i<nums.length-2;i++){
                if(nums[i]>0)//如果num[i]>0，没有循环的必要了，因为三个数中的最小值已经大于零。
                    break;
                if(i==0 || (i>0 && nums[i]!=nums[i-1])){
                    int j=i+1;
                    int k=nums.length-1;
                    while(j<k){
                        if(nums[i]+nums[j]+nums[k]>0)
                            k--;
                        else if(nums[i]+nums[j]+nums[k]==0){
                        //if(!result.contains(Arrays.asList(new Integer[]{nums[i],nums[j],nums[k]}))){
                            result.add(Arrays.asList(new Integer[]{nums[i],nums[j],nums[k]}));
                        //}
                            while(j<k && nums[j]==nums[j+1]) j++;
                            while(j<k && nums[k]==nums[k-1]) k--;
                            j++;k--;
                        }
                        
                        else
                            j++;
                    }
                }
                
            }
      
        return result;
        
    }
    public int findMiddle(int start,int end,int[] nums){
		int middle=(end+start)/2;
		if((middle-start==1 && nums[start]*nums[middle]<=0) || middle==start)
			return start;
		else{
			if(nums[start]*nums[middle]>0)
				return findMiddle(middle, end,nums);
			else
				return findMiddle(start, middle, nums);
		}
	}
}

主要的问题是消除重复解，因此代码中会判断nums[i]和nums[i-1]，以及nums[j]和nums[j+1]，以及nums[k]和nums[k+1]是否相等，如果相等则跳过该元素。如果nums[i]的值大于零，即可结束循环，因为三个数中的最小值已经大于零。