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

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

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

考虑输出需要,这里直接排序同时方便计算
注意去重问题

import java.util.*;

public class Solution {
    public ArrayList<ArrayList<Integer>> threeSum(int[] num) {
        Arrays.sort(num);
        ArrayList<ArrayList<Integer>> result=new ArrayList<ArrayList<Integer>>();
        //我再负数区 选一个则,必须在 正数区选一个,或者选两个
        for(int i=0;i<num.length;i++){
            if(num[i]>0){
                //非正数选完代表选完
                break;
            }
            if(i!=0&&num[i]==num[i-1]){
                //去重
                continue;
            }
            //两数求和

            int start=i+1;
            int end=num.length-1;
            while(start<end){
                //先左移 因为排序原因 start 为 相对于 i 第二小值
                while(start<end&&num[i]+num[start]+num[end]>0){
                    end--;
                }
                if(start<end&&num[i]+num[start]+num[end]==0){
                     ArrayList<Integer> item=new ArrayList<Integer>();
                     item.add(num[i]);
                     item.add(num[start]);
                     item.add(num[end]);
                     result.add(item);
                     //右移去重
                    while(start<end&&item.get(1)==num[start]){
                        start++;
                    }
                }else{
                    //右移增大整体数
                    start++;
                }
            }

        }

        return result;
    }
}
全部评论

相关推荐

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