题目描述

给定n个整数的一个数组S，S中是否有元素a、b和c满足a+b+c=0？找出数组中所有满足加和为0的不同的三个数组合。

注意，(a,b,c)中的元素必须是非降序的排列方式(即，a<=b<=c)

解决方案中给出的集合不能包含重复的三元组。

输入实例

例如，给定数组S={-1 0 1 2 -1 4}

输出实例

一个解决方案集合是：
(-1, 0,1)
(-1,-1,2)

代码示例

java代码如下所示：

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Set;


public class threeNum {

    public static void main(String[] args) {
        int[] S = {-1, 0, 1, 2, -1, -4,-3,-4,4,3,8};
        new threeNum().get3Sum(S);
    }

    public Set<String> get3Sum(int[] S){

        if(S.length<3 || S==null){
            return null;
        }

        //接收拼接的字符串
        StringBuffer sb = new StringBuffer();
        for(int i=0; i<S.length; i++){
            for(int j=0; j<S.length; j++){
                for(int z=0; z<S.length; z++){
                    //筛选出不是递减的一组元素
                    if(S[i]<=S[j] && S[j]<=S[z]){
                        int sum = S[i] + S[j] + S[z];
                        if(sum==0){
                            String str = "("+S[i]+","+S[j]+","+S[z]+")";    
                            sb.append(str+";");
                        }
                    }
                }
            }
        }

        String s = sb.toString();
        s = s.substring(0, sb.length()-1);
        String[] arr = s.split(";");

        Set<String> set = new HashSet<String>();
        //将所筛选出来的元素放入Set集合中，去重
        for (int k = 0; k < arr.length; k++) {
            set.add(arr[k]);
        }
        System.out.println(set);
        return set;
    }

    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new LinkedList<>();

        if (nums != null && nums.length > 2) {
            // 先对数组进行排序
            Arrays.sort(nums);
            // i表示假设取第i个数作为结果
            for (int i = 0; i < nums.length - 2; ) {
                // 第二个数可能的起始位置
                int start = i + 1;
                // 第三个数可能是结束位置
                int end = nums.length - 1;

                while (start < end) {
                    // 如果找到满足条件的解
                    if (nums[start] + nums[end] == -nums[i]) {
                        // 将结果添加到结果含集中
                        List<Integer> list = new ArrayList<>(3);
                        list.add(nums[i]);
                        list.add(nums[start]);
                        list.add(nums[end]);
                        result.add(list);

                        // 移动到下一个位置，找下一组解
                        end--;
                        start++;

                        // 从左向右找第一个与之前处理的数不同的数的下标
                        while (start < end && nums[start] == nums[start - 1]) {
                            start++;
                        }
                        // 从右向左找第一个与之前处理的数不同的数的下标
                        while (start < end && nums[end] == nums[end + 1]) {
                            end--;
                        }
                    }
                    // 和大于0
                    else if (nums[start] + nums[end] > -nums[i]) {
                        end--;
                        // 从右向左找第一个与之前处理的数不同的数的下标
                        while (start < end && nums[end] == nums[end + 1]) {
                            end--;
                        }
                    }
                    // 和小于0
                    else {
                        start++;
                        // 从左向右找第一个与之前处理的数不同的数的下标
                        while (start < end && nums[start] == nums[start - 1]) {
                            start++;
                        }
                    }
                }

                // 指向下一个要处理的数
                i++;
                // 从左向右找第一个与之前处理的数不同的数的下标
                while (i < nums.length - 2 && nums[i] == nums[i - 1]) {
                    i++;
                }
            }
        }

        return result;
    }


}

代码运行结果