找出数组中三个数之和为0的组合

• • 题目
  • 思路
  • 代码

题目

1. 给定一个无序可重复整数序列，当该序列中任意三个数的和等于0，输出这三个数。如：序列nums=[-1,0,1,2,-1,-4]，输出[[-1,0,1],[-1,-1,2]]

思路

首先对数组不同位进行两两结合，在进行一轮循环，看刚才结合的两组相加和不同位的数之和是否等于0

代码

public class ThreeNumbersSum { 

    public static void main(String[] args) { 
        int[] nums = new int[]{ -1, 0, 1, 2, -3, -4};
        List<int[]> ts = ts(nums);
        ts.forEach(arr -> { 
            System.out.println(Arrays.toString(arr));
        });
    }

    public static List<int[]> ts(int[] nums) { 

        if (nums == null || nums.length < 3) { 
            return new ArrayList<>();
        }
        int length = nums.length;
        List<int[]> resultList = new ArrayList<>();
        if (length == 3) { 
            if (nums[0] + nums[1] + nums[2] != 0) { 
                return null;
            } else { 
                resultList.add(nums);
                return resultList;
            }
        }

        // 将数组中不同位置的数进行求和，并记录位置
        List<Temp> maps = new ArrayList<>();
        for (int i = 0; i < length - 1; i++) { 
            for (int j = i + 1; j < length; j++) { 
                Temp temp = new Temp();
                temp.index[0] = i;
                temp.index[1] = j;
                temp.sum = nums[i] + nums[j];
                maps.add(temp);
            }
        }
        // 将第三个数和刚才那两个数之和相加，看是否等于0
        for (int i = 0; i < length; i++) { 
            for (Temp temp : maps) { 
                int[] index = temp.index;
                if (index[0] >= i || index[1] >= i) { 
                    continue;
                }
                if (temp.sum + nums[i] == 0) { 
                    temp.index[2] = i;
                    resultList.add(new int[]{ nums[index[0]], nums[index[1]], nums[i]});
                }
            }
        }

        return resultList;
    }
}
class Temp { 

    int[] index = new int[3];

    int sum;
}