[leetcode] Two Sum

使用一个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;
    }
}