15.3 Sum

3 Sum

题目

给定一个具有n个整数的数组S，问在S中是否存在这样的元素a, b, c满足 a + b + c = 0？找到数组中所有的满足和为零的特有三元素。
注：答案集中不能包含重复的三元素。

比如，给定数组 S = [-1, 0, 1, 2, -1, -4],
答案集为：
[
[-1, 0, 1],
[-1, -1, 2]
]

代码块

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
            List<List<Integer>> result = new ArrayList<List<Integer>>(); //建立一个列表的列表
        Arrays.sort(nums);//排序
        int i = 0 ;
        for( i = 0; i < nums.length - 2; i++){
            if(i != 0 && nums[i] == nums[i-1]){

                continue; //去除重复的元素
            }
            int j = i + 1;
            int k = nums.length - 1;
            while(j < k && j< nums.length - 1){
                List<Integer> temp = new ArrayList<Integer> ();
                if(nums[i] + nums[j] + nums[k] < 0){
                    j++;
                }else if(nums[i] + nums[j] + nums[k] > 0){
                    k--;
                }else{
                    temp.add(nums[i]);
                    temp.add(nums[j]);
                    temp.add(nums[k]);
                    result.add(temp);
                    j++;
                    k--;
                    while(j < k && nums[j] == nums[j-1]){
                        j++; //j与它前一个元素相等否
                    }
                    while(j < k && nums[k] == nums[k+1]){
                        k--;//k与后一个元素相等否
                    }   
                }

            }
        }
        return result;
    }
}

代码分析

第一步，对数组进行排序；（java中具有自动排序功能，可以直接调用Array**s** .sort 记着加s ，然后进行补全）
第二步，进行查找。再看i是否满足条件。
进入循环：length-2 是为了保证j,k不越界（i控制条件是与前一个元素不相等）；
然后给j,k 赋初值（j=i+1;k=length-1）。
满足第二层循环条件的话（j 是否满足条件j＜k,j小于length-1），
- 1.再将j后移（小于0）；
- 2.k前移（大于0）；
- 3.相等的话添加到列表内，并进行jk移位，以及去重。
第三步，返回结果。

因为使用了排序，以及两个循环。所以算法复杂度，O(nlogn+n^2)→O(n^2)。