题解 | #数组中相加和为0的三元组#

数组中相加和为0的三元组

http://www.nowcoder.com/practice/345e2ed5f81d4017bbb8cc6055b0b711

  1. 左右指针分别在两侧,向中心靠拢
  2. 使用set去重
  3. 如果等于0,在当前i循环下,不应该直接break,可能存在其他的组合,所有l++,r--,继续遍历可能存在的组合
  4. 返回的list严格遵守升序,所以要进行一次排序。

import java.util.*;

public class Solution {
   public ArrayList<ArrayList<Integer>> threeSum(int[] num) {
        Arrays.sort(num);
        Set<ArrayList<Integer>> set = new HashSet<>();

        for (int i = 1; i < num.length - 1; i++) {
            int begin = 0;
            int end = num.length - 1;
            while(begin < i && i < end) {
                int sum = num[i] + num[begin] + num[end];
                if (sum == 0) {
                    ArrayList<Integer> tmp = new ArrayList<Integer>();
                    tmp.add(num[begin]);
                    tmp.add(num[i]);
                    tmp.add(num[end]);
                    set.add(tmp);
                    begin++;
                    end--;
                }else if (sum > 0) {
                    end--;
                }else {
                    begin++;
                }
            }
        }
        ArrayList<ArrayList<Integer>> list = new ArrayList<>(set);
        list.sort((o1, o2) -> {
            for (int i = 0; i < o1.size(); i++) {
                if (o1.get(i) == o2.get(i)) {
                    continue;
                }
                if (o1.get(i) < o2.get(i)) {
                    return -1;
                }else {
                    return 1;
                }
            }
            return 0;
        });
        return list;
    }
}
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务