leetcode 1. Two Sum (Java)

题目

Given an array of integers, return indices of the two numbers such that they add up to a specific target.

You may assume that each input would have exactly one solution, and you may not use the same element twice.

难度

easy(做出来并不难，但要找一个好一点的解法)

思路

找一个O(n)的解法

1. 思路转换：两个数之和=target-> 第二个数 = target-第一个数
2. 因为既需要比较第二个数与target-第一个数的大小，又需要找出这个数所在的位置，所使用一个map来存储，key为数组中该项的值
3. 遍历数组，如果nums[i]不在map的key中，map.put(nums[i],i)；如果在，就返回这两个数的序列

代码

class Solution {
    public int[] twoSum(int[] nums, int target) {
        int s=nums.length;
        Map<Integer, Integer> map = new HashMap<>();
        for(int i=0;i<s;i++){
            if(map.containsKey(target-nums[i])){
            	//使用map.get获取找到的那个数的位置
                return new int[]{map.get(target-nums[i]),i};
            }else{
                map.put(nums[i],i);
            }
        }
        return new int[2];
    }
}

注意点

• 关于key和value，因为要比较的数的大小，所以nums[i]作为key, i作为value
• map存储的key相当于是当前找不到2个符合要求数的集合