当前位置：首页 > article >正文

BM50-两数之和

article 2024/9/25 7:13:39

题目

给出一个整型数组 numbers 和一个目标值 target，请在数组中找出两个加起来等于目标值的数的下标，返回的下标按升序排列。

（注：返回的数组下标从1开始算起，保证target一定可以由数组里面2个数字相加得到）

数据范围：2≤len(numbers)≤10^5，−10≤numbersi≤10^9，0≤target≤10^9。

要求：空间复杂度 O(n)，时间复杂度 O(nlogn)。

示例1

输入：[3,2,4],6

返回值：[2,3]

说明：因为 2+4=6 ，而 2的下标为2 ， 4的下标为3 ，又因为下标2 < 下标3 ，所以返回[2,3]。

示例2

输入：[20,70,110,150],90

返回值：[1,2]

说明：20+70=90。

思路

使用Map来降低时间复杂度，遍历数组，如果没有（target - 当前值）就将当前数字存入哈希表，如果有，返回该数字下标即可。

哈希表可以优化第二遍循环中对元素的检索速度。

代码

import java.util.*;

public class Solution {
    /**
     * 
     * @param numbers int整型一维数组 
     * @param target int整型 
     * @return int整型一维数组
     */
    public int[] twoSum (int[] numbers, int target) {
        HashMap<Integer, Integer> map = new HashMap<>();
        //遍历数组
        for(int i = 0; i < numbers.length; i++) {
            //将不包含target - numbers[i]装入map中，包含的话直接返回下标
            if(map.containsKey(target - numbers[i])) {
                return new int[]{map.get(target - numbers[i]) + 1, i + 1};
            } else {
                map.put(numbers[i], i);
            }
        }
        throw new IllegalArgumentException("No solution");
    }
}