[面試題]在數組中找出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]