[面试题]在数组中找出3个数使得它们和为0

给定一个数组S，试找出3个数a, b, c，使得a+b+c=0。也即从集合中找出所有的和为0的3个数。

例如：集合S={-1，0， 1， 2， -1， 4}，则满足条件的3个数有2对：(-1, 0, 1)和(-1, 2, -1)。注意（-1，1，0）与（-1，0，1）算同一个解，所以不用重复考虑。

当然该例子集合的解也可以写成:(0, 1, -1)和(2, -1, -1)。

 

参考了：http://blog.csdn.net/wangran51/article/details/8858398，他给的代码稍微有些问题。

 

使用Java代码重写算法

package com.sum3;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class Sum3 {

    private static Set<List<Integer>> fing_sum3(List<Integer> arr){
        Collections.sort(arr);//先对数组进行排序
        for(Integer aInteger : arr){
            System.out.println(aInteger);
        }
        List<Integer> list3 ;
        Set<List<Integer>> setList = new HashSet<>();
        
        for(int i = 0; i < arr.size()-2; i++){
            int j = i+1;
            int k = arr.size() - 1;
            while(j < k){
                //先固定arr[i]不动，左右一定逼近;arr[j]太小，往前移动一位
                if (arr.get(i) + arr.get(j) + arr.get(k) < 0 ) {
                    j++;
                } else if(arr.get(i) + arr.get(j) + arr.get(k) > 0){
                    k--;
                }else {
                    list3 = new ArrayList<>();
                    list3.add(arr.get(i));
                    list3.add(arr.get(j));
                    list3.add(arr.get(k));
                    setList.add(list3);
                    //j++;//这个需要删掉，不然的话有些元素没有被计算
                    k--;
                }
            }
        }
        return setList;
    }
    
    
    public static void main(String[] args) {
        Integer[] list = {2,1,0,-1,1,4,2,-2,-3};
        ArrayList<Integer> arrayList = new ArrayList<>(list.length);
        Collections.addAll(arrayList, list);
        Set<List<Integer>> output = fing_sum3(arrayList);
        for(List<Integer> list2 : output){
            System.out.println(list2);
        }
    }

}

输出结果：

-3 -2 -1 0 1 1 2 2 4

[-3, -1, 4]
[-3, 1, 2]
[-2, 1, 1]
[-2, 0, 2]
[-1, 0, 1]