18. 4Sum

二刷。

还是部分遍历+滑窗。

难点在于如何跳过重复的元素。

滑窗过程中遇到重复元素,和前2个固定坐标遇到重复元素是不一样的。
So u shall handle them in different ways respectively.

滑窗过程中,一旦判定成功,左右坐标的下一个如果相同,都没有判断的必要,所以要往二者之间不停地滑。“小心地滑”(beware of the slippery floor) 标识牌翻译成 slide cautiously

固定坐标遇到和上一个相等也要滑。。

public class Solution 
{
    public List<List<Integer>> fourSum(int[] nums, int target) 
    {
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        if(nums.length < 4) return res;
        Arrays.sort(nums);
        
        for(int i = 0; i < nums.length-3;i++)
        {
            for(int j = i+1; j < nums.length-2;j++)
            {
                int L = j+1;
                int R = nums.length-1;
                int total = nums[i] + nums[j];
                while(L < R)
                {
                    if(total + nums[L] + nums[R] > target) R--;
                    else if(total + nums[L] + nums[R] < target) L++;
                    else
                    {
                        List<Integer> tempList = new ArrayList<>();
                        tempList.add(nums[i]);tempList.add(nums[j]);tempList.add(nums[L]);tempList.add(nums[R]);
                        res.add(new ArrayList<>(tempList));
                        while(L < R && nums[L] == nums[L+1]) L++;
                        while(L < R && nums[R] == nums[R-1]) R--;
                        R--;L++;
                    }
                }
                while(j < nums.length-2 && nums[j] == nums[j+1]) j++;
            }
            while(i < nums.length -1 && nums[i] == nums[i+1]) i++;
        }
        return res;
    
    }
}

二刷。

固定2个,然后夹逼。

注意去重,然后别忘了一开始排序。。。

Time: O(nlgn)排序 + O(n³)
Space: O(n)

public class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> res = new ArrayList<>();
        if (nums.length < 4) return res;
        
        Arrays.sort(nums);
        
        for (int i = 0; i < nums.length - 3; i++) {
            for (int j = i + 1; j < nums.length - 2; j++) {
                int left = target - nums[i] - nums[j];
                int l = j + 1;
                int r = nums.length - 1;
                
                while (l < r) {
                    if (nums[l] + nums[r] < left) {
                        l ++;
                    } else if (nums[l] + nums[r] > left) {
                        r --;
                    } else {
                        List<Integer> tempList = new ArrayList<>();
                        tempList.add(nums[i]);
                        tempList.add(nums[j]);
                        tempList.add(nums[l ++]);
                        tempList.add(nums[r --]);
                        res.add(tempList);
                        while (l < r && nums[l] == nums[l-1]) {
                            l ++;
                        }
                        while (l < r && nums[r] == nums[r+1]) {
                            r --;
                        }
                    }
                }
                while (j < nums.length - 1 && nums[j] == nums[j+1]) {
                    j ++;
                }
                
            }
            while (i < nums.length - 1 && nums[i] == nums[i+1]) {
                i ++;
            }
        }
        
        return res;
    }
}
posted @ 2016-12-10 23:51  哇呀呀..生气啦~  阅读(144)  评论(0编辑  收藏  举报