用map解决leetcode_1_两数之和-进阶

1. 两数之和

题目描述：

提示

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

你可以假设每种输入只会对应一个答案，并且你不能使用两次相同的元素。

你可以按任意顺序返回答案。

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]

示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

提示：

• 2 <= nums.length <= 104
• -109 <= nums[i] <= 109
• -109 <= target <= 109
• 只会存在一个有效答案

进阶：你可以想出一个时间复杂度小于 O(n2) 的算法吗？

题解和代码：

package day02;

import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

public class leetcode_1_两数之和 {

    public static void main(String[] args) {
        // 创建Scanner对象以读取控制台输入
        Scanner sc = new Scanner(System.in);
        
        // 定义一个整数数组，这里假设数组是预先定义的
        int[] arr = { 2, 7, 11, 15 };
        
        // 从控制台读取目标和target，即两个数相加的目标值
        int target = sc.nextInt();
        
        // 调用twoSum方法来查找数组中哪两个数相加等于target，并返回这两个数的索引
        int[] result = twoSum(arr, target);
        
        // 输出这两个数的索引位置
        // 注意数组索引从0开始，所以输出的索引是0-based的
        System.out.println(result[0] + " " + result[1]);
    }

    /**
     * 查找数组中两个数的索引，使得这两个数的和为目标值target。
     *
     * @param nums   整数数组
     * @param target 目标和
     * @return 包含两个整数的数组，分别是这两个数的索引
     */
    public static int[] twoSum(int[] nums, int target) {
        // 初始化结果数组，用来存储找到的两个数的索引
        int[] res = new int[2];

        // 创建一个HashMap来存储数组元素及其对应的索引
        // key: 数组中的元素值
        // value: 数组中的元素索引
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();

        // 遍历数组中的每一个元素
        for (int i = 0; i < nums.length; i++) {
            // 计算当前元素与目标和之间的差值
            // 即我们需要找到的另一个数，使两者相加等于target
            int s = target - nums[i];
            
            // 如果差值存在于Map中，说明我们已经找到了一个有效的配对
            // 因为我们之前没有遇到过当前元素，所以这个配对一定是新的
            if (map.containsKey(s)) {
                // 设置结果数组的第一个索引为当前遍历到的索引
                res[0] = i;
                
                // 设置结果数组的第二个索引为差值所对应的索引
                // 这个索引是我们之前遍历时记录的
                res[1] = map.get(s);
                
                // 一旦找到匹配的一对，就返回结果
                return res;
            }
            
            // 如果当前元素还未出现在map中，则将其索引加入map
            // 这样在后续的遍历中可以快速查找是否有符合条件的另一数
            map.put(nums[i], i);
        }
        
        // 如果没有找到满足条件的两个数，则返回默认的结果数组
        // 在本例中，这个情况实际上不会发生，因为我们假设一定有解
        return res;
    }
}