Leetcode JAVA刷刷站（1）两数之和

news2024/11/15 6:50:13

一、题目概述

二、思路方向

为了解决这个问题，你可以使用哈希表（在Java中通常使用HashMap）来存储遍历过的数组元素及其对应的索引。这样，当你遍历数组时，你可以检查target - 当前元素是否已经在哈希表中，如果是，则找到了两个数的和为目标值的整数，并返回它们的索引。

三、代码实现

import java.util.HashMap;  
import java.util.Map;  
  
public class Solution {  
    public int[] twoSum(int[] nums, int target) {  
        // 创建一个哈希表来存储遍历过的元素及其索引  
        Map<Integer, Integer> map = new HashMap<>();  
          
        // 遍历数组  
        for (int i = 0; i < nums.length; i++) {  
            // 计算差值  
            int complement = target - nums[i];  
              
            // 检查差值是否已经在哈希表中  
            if (map.containsKey(complement)) {  
                // 如果在，则返回这两个数的索引  
                return new int[]{map.get(complement), i};  
            }  
              
            // 如果差值不在哈希表中，则将当前元素及其索引存入哈希表  
            map.put(nums[i], i);  
        }  
          
        // 如果没有找到这样的两个数，则抛出异常（但根据题目描述，这里不需要）  
        // 在实际应用中，可以根据需求进行错误处理或返回特殊值  
        throw new IllegalArgumentException("No two sum solution");  
    }  
  
    public static void main(String[] args) {  
        Solution solution = new Solution();  
        int[] nums = {2, 7, 11, 15};  
        int target = 9;  
        int[] result = solution.twoSum(nums, target);  
        System.out.println("Index1: " + result[0] + ", Index2: " + result[1]);  
    }  
}

执行结果：

四、小结

在这个实现中，我们遍历数组nums，对于每个元素nums[i]，我们计算它与目标值target的差值complement。然后，我们检查complement是否已经在哈希表map中。如果是，说明我们找到了两个数的和为目标值，直接返回这两个数的索引（map.get(complement)和i）。如果complement不在哈希表中，我们就将当前元素nums[i]及其索引i存入哈希表，以便后续检查。

注意，在题目描述中提到了“数组中同一个元素在答案里不能重复出现”，这个实现自动满足了这一要求，因为我们总是寻找当前元素之外的另一个元素来凑成目标值。