使用一个hashMap把遍历过的数组元素存起来,然后再遍历下一个元素的时候,只需要判断目标数减去这个数是否存在就好了。比如nums=[3,2,4] target=6.
遍历到3的时候,查找6-3是否出现在map里,不存在的话,把3写入map,遍历2的时候查找4是否存在,这时候4还没有放在map里,所以是不存在的。把2放在
map里。遍历4的时候,2就存在了,结果就找到了。
public class Solution {
public int[] twoSum(int[] nums, int target) {int len=nums.length;
Map<Integer,Integer> numMap=new HashMap<Integer,Integer>();
int other=0;
Integer des=0;
int[] ans=new int[2];
for(int i=0;i<len;++i){
other=target-nums[i];
des=numMap.get(other);
if(des!=null)
return searchIndex(nums,nums[i],other);
else
numMap.put(Integer.valueOf(nums[i]),1);
}
return ans;
}
private int[] searchIndex(int[] nums,int one,int other){
int[] des=new int[2];
int j=0,len=nums.length;
for(int i=0;i<len;++i){
if(nums[i]==one||nums[i]==other){
des[j++]=i+1;
if(j==2)
break;
}
}
return des;
}
}
优化代码,把searchIndex函数去掉,map的值直接存储Index的值。
public class Solution {
public int[] twoSum(int[] nums, int target) {
int len=nums.length;
Map<Integer,Integer> numMap=new HashMap<Integer,Integer>();
int other=0;
Integer des=0;
int[] ans=new int[2];
for(int i=0;i<len;++i){
other=target-nums[i];
des=numMap.get(other);
if(des!=null)
return searchIndex(nums,nums[i],other);
else
numMap.put(Integer.valueOf(nums[i]),other);
}
return ans;
}
private int[] searchIndex(int[] nums,int one,int other){
int[] des=new int[2];
int j=0,len=nums.length;
for(int i=0;i<len;++i){
if(nums[i]==one||nums[i]==other){
des[j++]=i+1;
if(j==2)
break;
}
}
return des;
}
}