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 感谢原博主!