leetcode 1. 两数之和

许我杂俎2024-02-28 11:03

cpp 复制代码

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        if(nums.size() == 2)
        return {0, 1};
        unordered_map <int, int> mapnums;
        for(int i = 0;i < nums.size();i++)
        {
            if(mapnums.count(target - nums[i]))
            {
               return {mapnums[target - nums[i]],i};
            }
            mapnums[nums[i]] = i;
        }
        return {-1,-1};
    }
};

输入验证:
- 首先，代码检查 nums 的大小是否为 2。如果是，则直接返回 {0, 1}。这是一个边界情况的处理，因为当数组只有两个元素时，它们必然是目标值的和。
使用哈希表:
- 接下来，代码创建了一个 unordered_map（无序映射）mapnums，用于存储数组中的每个元素及其对应的索引。
遍历数组:
- 使用一个 for 循环遍历数组 nums。
- 在每次迭代中，检查 mapnums 是否包含 target - nums[i] 这个键。这是基于这样一个事实：如果数组中存在两个数 a 和 b，使得 a + b = target，那么对于数组中的任意一个数 a，另一个数 b 必然是 target - a。
找到匹配项:
- 如果在 mapnums 中找到了 target - nums[i] 这个键，说明找到了两个数的和为 target。此时，返回这两个数的索引：mapnums[target - nums[i]]（target - nums[i] 的索引）和 i（当前遍历到的 nums[i] 的索引）。
更新哈希表:
- 如果在 mapnums 中没有找到 target - nums[i]，则将当前元素 nums[i] 和其索引 i 添加到 mapnums 中，以便在后续的迭代中进行查找。
返回默认结果:
- 如果循环结束后仍然没有找到和为 target 的两个数，则返回 {-1, -1}，表示没有找到这样的两个数。
这个算法的时间复杂度是 O(n)，其中 n 是数组 nums 的大小。因为只需要遍历一次数组，而每次查找哈希表的操作都是 O(1) 的时间复杂度。这个算法非常高效，特别是当数组很大时。