LeetCode-两数之和

题目

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。

示例:

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

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

https://leetcode-cn.com/problems/two-sum

方案

采用HashMap做中间状态记录,时间复杂度O(n), 空间复杂度O(n)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
class Solution {
    public int[] twoSum(int[] nums, int target) {
        int len = nums.length;
        HashMap<Integer, Integer> tempMap = new HashMap<>(len);
        for(int i=0; i<len; i++){
            if (tempMap.containsKey(nums[i])) {
                return new int[]{tempMap.get(nums[i]), i};
            } else {
                tempMap.put(target - nums[i], i);
            }
        }
        return null;
    }
}