LeetCode - 1. 两数之和

描述

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-sum/
 

求解

    class Solution {
    public:
        // 方法一，暴力解法，O(N * N)
        vector<int> twoSum_1e(vector<int> &nums, int target) {
            const int n = nums.size();
            for (int i = 0; i < n; ++i) {
                auto it = std::find(nums.cbegin() + i + 1, nums.cend(), target - nums[i]);
                if (it != nums.cend()) {
                    return vector<int>{i, int(it - nums.cbegin())};
                }
            }
            return vector<int>{-1, -1}; // 无解返回[-1，-1]，当然，题目是保证有解的，所以永远不会走这个分支
        }

        // 方法二，查找表, O(N)
        vector<int> twoSum(vector<int> &nums, int target) {
            const int n = nums.size();
            std::unordered_map<int, int> record; // <数值，位置>映射
            for (int i = 0; i < n; ++i) {
                if (record.count(target - nums[i]) > 0) {
                    return vector<int>{i, record[target - nums[i]]};
                }
                record[nums[i]] = i;
            }
            return vector<int>{-1, -1}; // 无解返回[-1，-1]，当然，题目是保证有解的，所以永远不会走这个分支
        }

    };