leetcode_15. 3Sum-求数组中三个数和为0

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note: The solution set must not contain duplicate triplets.

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]
把所有满足条件的三元组都要列出来!

思路:先排序,比如先把第一个元素定为nums[i]时,让后面两个元素和为(-1)*nums[i],转化为两元素和的问题。
遇到相同的元素就跳过去!

    public List<List<Integer>> threeSum(int[] nums) {

        List<List<Integer>> my = new ArrayList<>();
        Arrays.sort(nums);
        for(int i=0;i<=nums.length-3;i++)
            if(nums[i]<=0){
                if(i>=1 && nums[i]==nums[i-1])
                    continue;               
                int j=i+1;
                int k=nums.length -1;
                while(j<k){
                    if(nums[i]+nums[j]+nums[k]==0)
                    {
                        List<Integer> one = new ArrayList<>();
                        one.add(nums[i]);
                        one.add(nums[j]);
                        one.add(nums[k]);
                        my.add(one);    
                        j++;
                        k--;
                        while(j<k && nums[j]==nums[j-1])
                            j++;
                        while(j<k && nums[k]==nums[k+1])
                            k--;
                    }

                    else {
                        boolean flag =true;
                        // 求有序数组中两个数的和为(-1)*nums[i]
                        while(flag){
                            while(j<k && nums[j]+nums[k]>(-1)*nums[i])
                                k--;
                            while(j<k && nums[j]+nums[k]<(-1)*nums[i])
                                j++;
                            if( j>=k || nums[j]+nums[k]==(-1)*nums[i])
                                flag = false;
                        }
                    }

                }   // while(j<k)


            }

        return my;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值