Leetcode 两数之和

两数之和

问题描述:

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

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

题目链接

暴力法求解:

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

时间复杂度为:O(n^2)
空间复杂度为:O(1)

上面的代码size()函数返回vector向量的元素个数

哈希表求解:

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        map<int,int> numPair; // 增加了额外的空间复杂度
        for(int i = 0 ; i < nums.size() ; i++){
            numPair.insert(pair<int,int>(nums[i],i));
        }
        for(int i = 0 ; i < nums.size() ; i++){
            // 第一个条件是判断是否存在另一个被加数
            // 第二个条件是排除相同索引的值
            if(numPair.count(target-nums[i])==1 && numPair[target - nums[i]] != i){
                return {i,numPair[target - nums[i]]};
            }
        }
        return {-1,-1};
    }
};

时间复杂度为:O(n)
空间复杂度为:O(n)

上面的insert()为map中添加元素的函数,count()为map中元素是否存在的函数,返回值只有0或者1,0代表不存在,1代表存在

你可能感兴趣的:(leetcode题集,leetcode)