LeetCode 三數(shù)之和 — 改進(jìn)解法

題目：給定一個(gè)包含 n 個(gè)整數(shù)的數(shù)組 nums，判斷 nums 中是否存在三個(gè)元素 a，b，c ，使得 a + b + c = 0 ？找出所有滿足條件且不重復(fù)的三元組。
注意：答案中不可以包含重復(fù)的三元組。
例如, 給定數(shù)組 nums = [-1, 0, 1, 2, -1, -4]，
滿足要求的三元組集合為：
[ [-1, 0, 1], [-1, -1, 2] ]

　最開(kāi)始做的解法是先將整個(gè)數(shù)組排序；然后遍歷前兩個(gè)數(shù)(a和b)的所有情況(n^2)；對(duì)于第三個(gè)數(shù)，則從剩余的數(shù)中（即從第二個(gè)數(shù)下一個(gè)位置開(kāi)始到末尾）利用二分查找出是否存在數(shù)字 -(a+b)即可。 復(fù)雜度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下標(biāo)讀過(guò)的數(shù)字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下標(biāo)讀過(guò)的數(shù)字mj.put(nums[j],1);int temp = -(nums[i]+nums[j]);if(bSearch(nums,j+1,nums.length-1,temp) == false) continue; //二分搜索j下標(biāo)之后的區(qū)間是否有數(shù)字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;} }

里面有兩個(gè)用了哈希的地方，所以時(shí)間復(fù)雜度應(yīng)該還要乘上一個(gè)常數(shù)K...(解數(shù)組相關(guān)的題感覺(jué)總有些依賴哈希的方法=_= ...)

---

　最近做了另一個(gè)數(shù)組區(qū)間相關(guān)的題目，受其解法的啟發(fā)，打算對(duì)這道題解法進(jìn)行優(yōu)化。
　還是先對(duì)數(shù)組進(jìn)行排序；對(duì)第一個(gè)數(shù)字(a)進(jìn)行遍歷，而然后在剩余的數(shù)中用前后指針的方法找出兩個(gè)和為-a的數(shù)字：兩個(gè)指針left和right；left初始化為數(shù)字a的下一位置，right為最后一個(gè)位置。比較nums[left]+nums[right]+a和0的大小；如果大于0則right--，小于就left++。 其中在添加結(jié)果集時(shí)需考慮去重問(wèn)題，用個(gè)哈希判斷是否有重復(fù)就行了，復(fù)雜度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下標(biāo)讀過(guò)的數(shù)字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;} }

轉(zhuǎn)載于:https://www.cnblogs.com/geek1116/p/10172313.html

總結(jié)

以上是生活随笔為你收集整理的LeetCode 三数之和 — 优化解法的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。