【leetcode hot 100】【1】1. 两数之和 【c++ rust go】

题目

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

解题思路

题目要求在一个整数数组中寻找两个数，使得它们的和等于给定的目标值，并返回它们的下标。

为了寻找这两个数，我们可以遍历整个数组，对于每个元素，都计算它的补数（即目标值减去当前元素的值），然后检查补数是否已经在之前的元素中出现过。我们可以使用哈希表（Python 中的字典，Go 中的 map，Rust 中的 HashMap）来快速查找补数是否在数组中出现过，同时还需要在哈希表中记录每个元素的下标，方便返回结果。

具体实现过程中，我们可以将元素和下标分别作为哈希表的键和值，这样就可以在哈希表中用补数来快速查找数组中是否已经出现过这个数，如果存在，就可以直接返回对应的下标。

代码

C++

#include <vector>
#include <unordered_map>

class Solution {
public:
    std::vector<int> twoSum(std::vector<int>& nums, int target) {
        std::unordered_map<int, int> hash_table; // 创建一个哈希表
        for (int i = 0; i < nums.size(); i++) {
            int complement = target - nums[i]; // 计算当前元素的补数
            if (hash_table.find(complement) != hash_table.end()) { // 如果补数存在于哈希表中，返回对应的索引
                return { hash_table[complement], i };
            }
            hash_table[nums[i]] = i; // 否则将当前元素和其索引加入哈希表中
        }
        return {}; // 没有找到符合条件的元素，返回空数组
    }
};

Go

func twoSum(nums []int, target int) []int {
    hash_table := make(map[int]int) // 创建一个 map，用于存储数组元素和它们的索引
    for i, num := range nums {
        complement := target - num // 计算当前元素的补数
        if j, ok := hash_table[complement]; ok { // 如果补数存在于 map 中，返回对应的索引
            return []int{j, i}
        }
        hash_table[num] = i // 否则将当前元素和其索引加入 map 中
    }
    return nil // 没有找到符合条件的元素，返回 nil
}

Rust

use std::collections::HashMap;
impl Solution {
    pub fn two_sum(nums: Vec<i32>, target: i32) -> Vec<i32> {
        let mut hash_table = HashMap::new(); // 创建一个哈希表
        for (i, num) in nums.iter().enumerate() {
            let complement = target - num; // 计算当前元素的补数
            if let Some(j) = hash_table.get(&complement) { // 如果补数存在于哈希表中，返回对应的索引
                return vec![*j as i32, i as i32];
            }
            hash_table.insert(num, i as i32); // 否则将当前元素和其索引加入哈希表中
        }
        vec![] // 没有找到符合条件的元素，返回空数组
    }
}