[leetcode] Two Sum

来源:互联网 发布:shell获取java版本号 编辑:程序博客网 时间:2024/04/28 12:44

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


0 0