15. 3Sum - medium

2016-10-20  本文已影响3人  沉睡至夏

注意要跳过重复。
把每个element 看成三个数中的一个,然后在这个数后面剩余的数里,从最大和最小的数开始做2sum.

public class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> ans = new ArrayList<List<Integer>>();
        
        for (int i=0; i<nums.length; i++) {
            // skip same elements:
            if(i==0 || (i>0 && nums[i] != nums[i-1])) {
                int lo = i+1, hi = nums.lengh-1, sum = 0-nums[i];
                // try sum:
                while (lo < hi) {
                    if (nums[lo] + num[hi] == sum) {
                        res.add(Arrays.asList(num[i], num[lo], num[hi]));
                        // skip same elements:
                        while (lo < hi && nums[lo] == nums[lo+1]) lo++;
                        while (lo < hi && nums[hi] == nums[hi-1]) hi--;
                        lo++; hi--;
                    }
                    else if (num[lo] + num[hi] < sum) lo++;
                    else  hi--;
                }
            }
        }
        return res;
    }
}
上一篇下一篇

猜你喜欢

热点阅读