题目:
给出一个有n个元素的数组S,S中是否有元素a,b,c满足a+b+c=0?找出数组S中所有满足条件的三元组。
注意:
- 三元组(a、b、c)中的元素必须按非降序排列。(即a≤b≤c)
- 解集中不能包含重复的三元组。
- 例如,给定的数组 S = {-10 0 10 20 -10 -40},解集为(-10, 0, 10) (-10, -10, 20)
解答:
- 首先要判断数组num是否为空
- 将数组排序,(排序有多种方法)
- 遍历数组,分析题目可知,三个数相加等于零,此时的数组num已经是排过序的,由小到大的顺序,所以第一个数字如果大于0的话,那么后面的数一定都大于0,所以此处可以加一个判断。第一个数字大于0点num,直接返回(后面的值如何相加都不可能等于0)
- 由于存在list1是一个三元组的形式,所以此处还需再设一个ArrayList来接收满足要求的三个数,再将这个集合放入list1中.
- 定义num[i],num[j],num[k]是我们需要找出的三个数字。遍历这个num数组,然后找出这三个数相加等于0.
代码如下:
import java.util.ArrayList;
import java.util.Arrays;
public class Solution {
public ArrayList<ArrayList<Integer>> threeSum(int[] num) {
ArrayList<ArrayList<Integer>> list1=new ArrayList<>();
if(num==null){
return list1;
}
Arrays.sort(num);
for(int i=0;i<num.length-2;i++){
if(num[i]>0){
break;
}
if(i!=0 && num[i]==num[i-1]){
continue;
}
for(int j=i+1;j<num.length-1;j++){
if(j!=i+1 && num[j]==num[j-1]){
continue;
}
for(int k=j+1;k<num.length;k++){
if(k!=j+1 && num[k]==num[k-1]){
continue;
}
int sum=num[i]+num[j]+num[k];
if(sum==0){
ArrayList<Integer> list=new ArrayList<>();
list.add(num[i]);
list.add(num[j]);
list.add(num[k]);
list1.add(list);
}
}
}
}
return list1;
}
}