15. 3Sum

来源:互联网 发布:邮件大师 mac 编辑:程序博客网 时间:2024/06/08 10:54

Given an array S of n integers, are there elements abc 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]] 

这道题的意思是:有一组数组,需要我们找到数组中的三个数字能够使这三个数字的和是0,用列表的形式返回这三个数字。首先对数组进行排序,然后可以用sun表示除了第一个数字外其他两个数字的和,然后用二分查找的方法去判断索引为lo的数字和索引为hi的数字的和与sum的大小,如果等于sum,则把这三个数字以列表的形式添加进列表中,并且lo增加1,hi减1,如果这两个数字的和大于sum,这时候因为数组有序,可以通过减小hi使nums[hi]减小,反之则增加lo。不过在整个for循环中要保证数字不重复,但是不能用nums[i]!=nums[i+1]控制,这样程序会直接跳过nums[i]。


List<List<Integer>> res = new ArrayList<List<Integer>>();        Arrays.sort(nums);                for(int i=0;i<nums.length-2;i++){        if(i==0||i>0&&nums[i]!=nums[i-1]){        int lo = i+1;        int hi = nums.length-1;        int sum = 0-nums[i];        //System.out.println(sum);        while(lo<hi){        if(nums[lo]+nums[hi]==sum){        res.add(Arrays.asList(nums[i],nums[lo],nums[hi]));            while(lo<hi&&nums[lo]==nums[lo+1]){lo++;}            while(lo<hi&&nums[hi]==nums[hi-1]){hi--;}            lo++;hi--;        }else if(nums[lo]+nums[hi]>sum){hi--;}        else lo++;          }            }        }        return res;


原创粉丝点击