leetcode1 两数之和

两数之和题解

背景

link
https://leetcode.cn/problems/two-sum/description/
本题是leetcode题库的第一题，作为一道入门级的题目，在解法上也是很有意思的，本文将尝试给出几种常见解法。

题目

给定一个整数数组 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]

解法

1. 两数之和，时间复杂度有限，考虑直接两层for循环解决问题，
   时间复杂度O(n^2)。

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

2. 作为一个leetcoder，有时候官方给出的时间和内存的排名还是能激起优化的欲望的。显然，优化的常规思路就是降复杂度，两层for循环如果能优化成一层，那么优化效果显而易见。我们每次选择一个数nums[i]，所要求得的另一个值则可以表示为target - nums[i]。解法1也正是运用了这种思想，不过它的选择是每次重新遍历一遍，这层的复杂度是O(n), 有没有一种方法能降一降呢？当然在我们选中一个值的同时，尝试通过数据结构的记录而非遍历去查找另一个值是个很好的思路。红黑树是一个不错的结构，可惜的是平均查找复杂度是O(lgn)。幸好，C++还提供了unordered_map, 不考虑数据的有效性的情况下能够实现O(1)的复杂度，这样我们的算法就能优化到O(n)，大功告成。

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> value;
        const int n = nums.size();
        for (int i = 0; i < n; i++) {
            int k = target - nums[i];
            if (value.count(k) != 0) {
                return {i, value[k]};
            } else {
                value[nums[i]] = i;
            }
        }
        return {};
    }
};