算法:数组中相加为0的三元组

给出一个有n个元素的数组S,S中是否有元素a,b,c满足a+b+c=0?找出数组S中所有满足条件的三元组。

注意:
三元组(a、b、c)中的元素必须按非降序排列。(即a≤b≤c)
解集中不能包含重复的三元组。

本问题有多种解法,此处用ArrayList的方法

  1. 首先要判断数组num是否为空
  2. 将数组排序,(排序有多种方法)
  3. 遍历数组,分析题目可知,三个数相加等于零,此时的数组num已经是排过序的,由小到大的顺序,所以第一个数字如果大于0的话,那么后面的数一定都大于0,所以此处可以加一个判断。
  4. 由于存在list1是一个三元组的形式,所以此处还需再设一个ArrayList来接收满足要求的三个数,再将这个集合放入list1中.代码如下:
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;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_46535360/article/details/108433553