手撸算法-两数之和-牛客

描述

给出一个整数数组，请在数组中找出两个加起来等于目标值的数，
你给出的函数twoSum 需要返回这两个数字的下标（index1，index2），需要满足 index1 小于index2.。注意：下标是从1开始的
假设给出的数组中只存在唯一解
例如：
给出的数组为 {20, 70, 110, 150},目标值为90
输出 index1=1, index2=2

示例1

输入：
[3,2,4],6
返回值：
[2,3]

特别注意:

• 返回的数组下标从1开始

思路:

• 1，遍历元素，当前遍历元素为num1，所查找元素为num2=target-num1，如果在Map中能查找到num2，返回num2的下标和当前num1的下标；如果找不到则将num1和其下标存入Map；

• 2，关键点：边遍历边存入Map，这样就不担心有重复的元素，如果事先把所有的元素存入Map，则要考虑重复的元素，会提升复杂度。

 public int[] twoSum (int[] numbers, int target) {
        // write code here
        Map<Integer,Integer> map = new HashMap<>();
       
        for (int i = 0; i < numbers.length; i++) {
            int num1 = numbers[i];
            int num2 = target - num1;
            if (!map.containsKey(num2)) {
                map.put(num1,i);
            } else {
                return new int[]{map.get(num2)+1,i+1};
            }
        }
        
        throw new IllegalArgumentException("No solution");
    }