程序员

LeetCode1.两数之和题解

2020-07-09  本文已影响0人  星期八我一定好好读书

题目描述

两数之和
给定一个整数数组 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;
        
    }
};
上一篇下一篇

猜你喜欢

热点阅读