Leetcode 题解系列--Leetcode1 两数之和

题目描述

两数之和
给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。

解题思路

解法一：直观的解法，时间复杂度$O(n^2)$

遍历每个数，然后从下一个位置开始第二次遍历，查找是否存在一个数与当前数的和等于target，如果找到则返回答案。否则将第一层遍历的下标往后挪一位。如果遍历完整个数组都没有找到符合条件的数据，则返回空

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

解法二：利用空间换取时间，时间复杂度O(n)，空间复杂度O(n)

利用hash表将遍历到的数字和下标存起来，每次遍历到某个数时，先判断target在不在hash中，如果存在，则找到一组答案；如果不在则将当前的数和下标存起来。如果遍历完整个数组还没有找到答案，则返回空

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