15. 3sum(Nsum总结)

如果不用hashmap解决2sum,而是把2sum看成nSum中的一部分,那么就需要用two pointers来解决问题。

2sum伪代码:

1. 先对给定的数组进行排序

2. walker指第一个元素,runner指最后一个元素

3.当walker小于runner:

  1)如果walker和runner的和等于target,那么:

    (1)如果只需要一对解,记录答案,break

    (2)如果需要所有解,那么walker++, runner--.

       同时,为了节省时间,如果walker<runner && nums[walker]==nums[walker-1]时,walker++,并且如果walker<runner&&nums[runner]==nums[runner=1]时,runner--

  2) else如果和小于target,walker++

  3) else runner++

4.返回结果

 

算法时间复杂度

1. 2sum的时间复杂度

排序的复杂度是O(NlogN), 然后再就是最多从头到尾走一遍,所以是O(N), 所以复杂度是O(N+NlogN),所以是O(N)

2. 3sum时间复杂度是

排序的时间复杂度O(NLogN), 在对于每一个数组做2sum,但是这个时候的2sum不需要排序,所以这个2sum是O(N),所以这部分的时间复杂度是O(N^2)

所以总时间复杂度是O(N^2+NLogN)= O(N^2) 

 

public class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        if(nums == null || nums.length < 3) {
            return res;
        }
        Arrays.sort(nums);
        for(int i = 0; i < nums.length - 2; i++) {
            //if(i != nums.length-3 && nums[i] == nums[i+1]) {
            //    continue;
            //}//不可以加因为比如[-1,-1,2]也是合法解
            twoSum(nums, -nums[i], i, res);
        }
        return res;
    }
    
    private void twoSum(int[] nums, int target, int start, List<List<Integer>> res) {
        int walker = start + 1;
        int runner = nums.length - 1;
        while(walker < runner) {
            int sum = nums[walker] + nums[runner];
            if(sum == target) {
                List<Integer> item = new ArrayList<Integer>();
                item.add(nums[start]);
                item.add(nums[walker]);
                item.add(nums[runner]);
                if(!res.contains(item)) {
                    res.add(item);
                }
                walker++;
                runner--;
                while(walker<runner && nums[walker] == nums[walker-1]) {
                    walker++;
                }//不加会run time error
                while(walker<runner && nums[runner] == nums[runner+1]) {
                    runner--;
                }//同理
            } else if(sum < target) {
                walker++;
            } else {
                runner--;
            }
        }
    }
}

 

 

参考链接: http://blog.csdn.net/doc_sgl/article/details/12462151#comments 感谢原博主!

posted @ 2016-01-06 19:34  warmland  阅读(256)  评论(0编辑  收藏  举报