天猫高频编程考题：两数之和

题目描述

给定一个整数数组 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 <=
• - <= nums[i] <=
• - <= target <=
• 只会存在一个有效答案

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

解题思路

这个问题可以使用哈希表（HashMap）来实现。我们可以通过一次遍历数组的方式解决该问题。具体步骤如下：

1. 创建一个哈希表：用来存储数组中已经访问过的元素及其对应的下标。
2. 遍历数组：对数组中的每一个元素，计算出它与目标值之间的差值 complement。
3. 查找补数：检查哈希表中是否存在这个差值。如果存在，说明我们已经找到了两个数，它们的和等于目标值，返回它们的下标。
4. 更新哈希表：如果当前元素的补数不在哈希表中，将当前元素和它的下标加入哈希表中。
5. 返回结果：当找到符合条件的两个数时，直接返回它们的下标。

复杂度分析

• 时间复杂度：O(n)。我们只需遍历数组一次，对于数组中的每个元素，哈希表的查找和插入操作的时间复杂度都是 O(1)，因此总的时间复杂度为 O(n)。

• 空间复杂度：O(n)。在最坏的情况下（没有两个元素的和为目标值），我们需要在哈希表中存储数组中所有的元素及其下标，因此空间复杂度为 O(n)。

代码实现

package org.zyf.javabasic.letcode.hot100.hash;

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

/**
 * @program: zyfboot-javabasic
 * @description: 两数之和
 * @author: zhangyanfeng
 * @create: 2024-08-21 20:16
 **/
public class TwoSumSolution {
    public int[] twoSum(int[] nums, int target) {
        // 创建一个哈希表来存储已经访问过的元素及其下标
        Map<Integer, Integer> numMap = new HashMap<>();

        // 遍历数组中的每一个元素
        for (int i = 0; i < nums.length; i++) {
            int complement = target - nums[i];

            // 检查补数是否在哈希表中
            if (numMap.containsKey(complement)) {
                // 如果补数在哈希表中，返回补数的下标和当前元素的下标
                return new int[] { numMap.get(complement), i };
            }

            // 如果补数不在哈希表中，将当前元素和下标加入哈希表
            numMap.put(nums[i], i);
        }

        // 按题意，只会存在一个有效答案，所以不需要额外的返回值
        throw new IllegalArgumentException("No two sum solution");
    }

    public static void main(String[] args) {
        TwoSumSolution solution = new TwoSumSolution();

        int[] nums1 = {2, 7, 11, 15};
        int target1 = 9;
        int[] result1 = solution.twoSum(nums1, target1);
        System.out.println("Result 1: [" + result1[0] + ", " + result1[1] + "]"); // 输出: [0, 1]

        int[] nums2 = {3, 2, 4};
        int target2 = 6;
        int[] result2 = solution.twoSum(nums2, target2);
        System.out.println("Result 2: [" + result2[0] + ", " + result2[1] + "]"); // 输出: [1, 2]

        int[] nums3 = {3, 3};
        int target3 = 6;
        int[] result3 = solution.twoSum(nums3, target3);
        System.out.println("Result 3: [" + result3[0] + ", " + result3[1] + "]"); // 输出: [0, 1]
    }
}

具体可参考：https://zyfcodes.blog.csdn.net/article/details/141401712