LeetCode 三数之和 — 优化解法

LeetCode 三数之和 — 改进解法

题目：给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。
注意：答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[ [-1, 0, 1], [-1, -1, 2] ]

　最开始做的解法是先将整个数组排序；然后遍历前两个数(a和b)的所有情况(n^2)；对于第三个数，则从剩余的数中（即从第二个数下一个位置开始到末尾）利用二分查找出是否存在数字 -(a+b)即可。 复杂度O(n^2·logn) ：

class Solution {public List<List<Integer>> threeSum(int[] nums) {List<List<Integer>> ans = new ArrayList<List<Integer>>();Arrays.sort(nums);Map<Integer,Integer> mi = new HashMap();for(int i=0;i<nums.length-1;i++) {if(mi.get(nums[i]) != null) continue; //记录i下标读过的数字mi.put(nums[i],1);Map<Integer,Integer> mj = new HashMap();    for(int j=i+1;j<nums.length;j++) {if(mj.get(nums[j]) != null) continue; //记录j下标读过的数字mj.put(nums[j],1);int temp = -(nums[i]+nums[j]);if(bSearch(nums,j+1,nums.length-1,temp) == false) continue; //二分搜索j下标之后的区间是否有数字tempans.add(Arrays.asList(nums[i],nums[j],temp));}}return ans;}//二分算法public boolean bSearch(int[] nums,int s,int e,int key) {int start=s,end=e,mid;while(start<=end){mid = (start+end)/2;if(key < nums[mid]) end=mid-1;else if(key > nums[mid]) start=mid+1;else if(key == nums[mid]) return true;}return false;}
}

里面有两个用了哈希的地方，所以时间复杂度应该还要乘上一个常数K...(解数组相关的题感觉总有些依赖哈希的方法=_= ...)

---

　最近做了另一个数组区间相关的题目，受其解法的启发，打算对这道题解法进行优化。
　还是先对数组进行排序；对第一个数字(a)进行遍历，而然后在剩余的数中用前后指针的方法找出两个和为-a的数字：两个指针left和right；left初始化为数字a的下一位置，right为最后一个位置。比较nums[left]+nums[right]+a和0的大小；如果大于0则right--，小于就left++。 其中在添加结果集时需考虑去重问题，用个哈希判断是否有重复就行了，复杂度O(n^2·K) ：

class Solution {public List<List<Integer>> threeSum(int[] nums) {List<List<Integer>> ans = new ArrayList<List<Integer>>();Arrays.sort(nums);Map<Integer,Integer> mi = new HashMap();for(int i=0;i<nums.length-2;i++) {if(mi.get(nums[i]) != null) continue; //记录i下标读过的数字mi.put(nums[i],1);Map<Integer,Integer> mj = new HashMap();int l = i+1, r = nums.length-1, temp;while(l<r) {temp = nums[i] + nums[l] + nums[r];if(temp < 0) {l++;} else if(temp > 0) {r--;} else {if((mj.get(nums[l])) == null) {ans.add(Arrays.asList(nums[i], nums[l], nums[r]));mj.put(nums[l], 1);}l++; r--;}}}return ans;}
}