LC1 哈希表

粥粥zyw

已于 2024-07-27 13:14:49 修改

阅读量211

点赞数 4

文章标签：算法 leetcode java

于 2024-07-27 13:14:06 首次发布

本文链接：https://blog.csdn.net/weixin_44959470/article/details/140734072

版权

两数之和算法优化

在编程面试中，“两数之和” 是一个经典的问题。本文将介绍如何使用哈希表优化这一问题的解决方案，从而提升算法的效率。

问题描述

给定一个整数数组 nums 和一个目标值 target，您需要找到两个不同的索引 i 和 j，使得 nums[i] + nums[j] = target。请注意，假设每个输入只对应一种答案，并且不允许使用相同的元素。

示例

输入:

nums = [2, 7, 11, 15], target = 9

输出:

[0, 1]

解释: nums[0] + nums[1] = 2 + 7 = 9，因此返回 [0, 1]。

优化算法思路

我们可以使用哈希表来优化这个问题的解决方案。具体步骤如下：

遍历数组并构建哈希表: 在第一次遍历中，将每个元素及其索引存入哈希表，以便在后续查找中快速访问。
计算差值并查找: 在第二次遍历中，计算目标值与当前元素的差值。如果该差值存在于哈希表中，并且其索引与当前元素的索引不同，则找到答案。

这种方法的优势在于可以将时间复杂度降低到 O(n)。

代码实现

以下是实现这一算法的 Java 代码：

import java.util.HashMap;

class Solution {
    public int[] twoSum(int[] nums, int target) {
        int[] result = new int[2];
        HashMap<Integer, Integer> map = new HashMap<>();
        
        // 第一次遍历：构建哈希表
        for(int i = 0; i < nums.length; i++){
            map.put(nums[i], i);
        }
        
        // 第二次遍历：查找目标差值
        for(int j = 0; j < nums.length; j++){
            int diff = target - nums[j];
            if(map.containsKey(diff) && map.get(diff) != j){
                result[0] = map.get(diff);
                result[1] = j;
                return result;
            }
        }
        
        return result; // 如果没有找到，返回默认值
    }
}

代码解析

导入 HashMap:
```
import java.util.HashMap;
```
引入 HashMap 类以支持哈希表功能。
初始化结果数组和哈希表:
```
int[] result = new int[2];
HashMap<Integer, Integer> map = new HashMap<>();
```
创建一个结果数组用于存储索引，并初始化哈希表。
构建哈希表:
```
for(int i = 0; i < nums.length; i++){
    map.put(nums[i], i);
}
```
遍历数组，将每个元素及其对应的索引存入哈希表。

查找目标差值:

for(int j = 0; j < nums.length; j++){
    int diff = target - nums[j];
    if(map.containsKey(diff) && map.get(diff) != j){
        result[0] = map.get(diff);
        result[1] = j;
        return result;
    }
}