刚刚写3sum不是很顺,关于越过重复数字的处理,没有设置好,最后还是通过先 ++/--,再做while(有效范围 && 重复情况) 来实现。。。
所以进而想到额2sum,于是看了下从前的代码和实现,之前的是两个遍历,复杂度是N平方,最naive的做法。。。我想了个sort的方法,结果出错了,一看原来是题目要求返回index,那就显然不能sort,sort打乱了原有index和elem的对应关系。。。然而我又想了办法就是用HashMap来存储<elem, index>这样的对应关系,遍历一遍,通过判断map中是否包含了(target-nums[i]) ,来确定是否找到了match的一对。。。这个方法很想maxSubArray那道题。。。
代码如下:
public class Solution {
public int[] twoSum(int[] nums, int target) {
// solution 1, O(N*N)
/* if(nums.length<=1) return new int[]{-1,-1};
for(int i=0; i<nums.length; i++){
for(int j=i+1; j<nums.length; j++)
if(nums[i]+nums[j]==target) return new int[]{i,j};
}
return new int[]{-1,-1};*/
// 错误理解题意了,题目让你返回index,排序就打乱的原有的index
/* Arrays.sort(nums);
int[] result = new int[2];
int left=0;
int right=nums.length-1;
while(left<right){
int sum=nums[left]+nums[right];
if(sum==target){
result[0]=left;
result[1]=right;
break;
}
else if(sum<target) left++;
else right--;
}
return result;*/
// solution 2, O(N), with HashMap
Map<Integer, Integer> map= new HashMap<Integer, Integer>();
int[] result = new int[2];
for(int i=0; i<nums.length; i++){
if(map.containsKey(target-nums[i])){
result[0]=map.get(target-nums[i]);
result[1]=i;
break;
}else map.put(nums[i], i);
}
return result;
}
}