Featured image of post One algorithm a day(两数之和)

One algorithm a day(两数之和)

leetcode算法题库第一道题

题目:两数之和

问题描述:

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍

**难度:**简单

示例:

1
2
3
4
给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

代码

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
public class Topic001 {
    public static void main(String[] args) {
        // 测试
        Topic001 topic001 = new Topic001();
        int[] nums = {2, 7, 11, 15};
        int[] result = topic001.toSum(nums, 9);
        System.out.println(Arrays.toString(result));
    }
    public int[] toSum(int[] nums, int target) {
        Map<Integer, Integer> map = new HashMap<>(16);
        for (int i = 0; i < nums.length; i++) {
            int key = target - nums[i];
            if (map.containsKey(key)) {
                return new int[]{map.get(key), i};
            }
            map.put(nums[i], i);
        }
        return null;
    }
}

思路:图解

用一个HashMap存储数据,key是数组的元素,value是该元素在数组中的索引。因为要用到数据里的元素去比较,成功还要返回对应的索引。所以数组元素以及索引都要存起来并且对应起来,hash操作又是常数操作,所以对数组元素求hash很快。所以使用HashMap。

图解topic1

Licensed under CC BY-NC-SA 4.0