LeetCode第一题Two Sum

一开始没想到HashMap，看了别人的答案之后才知道用hashmap更快捷，第一次的代码如下：

class Solution {
    public int[] twoSum(int[] nums, int target) {
        int[] result = new int[2];
        Map<Integer,Integer> map = new HashMap<>();
        for(int i=0;i<nums.length;i++){
            map.put(nums[i],i);
        }
        for(int i=0;i<nums.length;i++){
            // 不能重复，再加一个判断条件
            if(map.containsKey(target-nums[i]) && map.get(target-nums[i])!=i){
                result[0]=i;
                result[1]=map.get(target-nums[i]);
                return result;
            }
        }
        return null;
        
    }
}

这样其实复杂度还不是最优的情况，因为有两个for循环，于是改变思路，只用一个for，代码如下：

class Solution {
    public int[] twoSum(int[] nums, int target) {
        int[] result = new int[2];
        Map<Integer,Integer> map = new HashMap<>();
        for(int i=0;i<nums.length;i++){
            if(map.containsKey(target-nums[i])){
                result[0]=map.get(target-nums[i]);
                result[1]=i;
                return result;
            }
            map.put(nums[i],i);
        }
        return result;
        
    }
}

以后尽量在需要使用查找的时候想一想能不能使用map来定位。