数组S中所有满足条件的三元组。

tech2024-08-04  66

题目描述 给出一个有n个元素的数组S,S中是否有元素a,b,c满足a+b+c=0?找出数组S中所有满足条件的三元组。 注意: 三元组(a、b、c)中的元素必须按非降序排列。(即a≤b≤c) 解集中不能包含重复的三元组。 例如,给定的数组 S = {-1 0 1 2 -1 -4},解集为(-1, 0, 1) (-1, -1, 2)

import java.util.*; public class Solution { public ArrayList<ArrayList<Integer>> threeSum(int[] num) { //-4,-1,-1,0,1,2 Arrays.sort(num); ArrayList<ArrayList<Integer>> h = new ArrayList<>(); for(int i=0;i<num.length; ++i) { if(i > 0 && (num[i]==num[i-1])) continue; int l=i+1,r = num.length-1; while(l<r) { int sum = num[l]+num[i] +num[r]; if(sum <0 ) { //left 过小 ++l; }else if(sum > 0) { //过大 --r; }else { ArrayList<Integer> m = new ArrayList<>(); m.add(num[i]);m.add(num[l]);m.add(num[r]); h.add(m); while((l<r)&& num[l]==num[l+1]) ++l; while(r> l && num[r]==num[r-1]) --r; ++l;--r; } } } return h; } }
最新回复(0)