LeetCode-1.两数之和-梦想新起航

最新推荐文章于 2024-09-09 18:27:53 发布

莽夫搞战术

最新推荐文章于 2024-09-09 18:27:53 发布

阅读量91

点赞数

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/yd778473278/article/details/119390889

版权

LeetCode 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

LeetCode-1.两数之和

题目

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

提示：
2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案
进阶：你可以想出一个时间复杂度小于 O(n2) 的算法吗？

题解

暴力解法（最简单）

不多说了两次for循环

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

提交结果
执行用时: 460 ms，超过10%
内存消耗: 9.8 MB，超过67%

速度优化

既然要速度优化那么我们就得想一个办法只遍历一遍数组。那么引入一个新的问题，怎么快速找到另一个数，同时获得这个数的位置。因此想到引入哈希表（字典，map）。

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> answer;
        map<int, int> candidate;
        for (unsigned int i = 0; i < nums.size(); i++) {
            int need = target - nums[i];
            map<int, int>::iterator iter = candidate.find(need);
            if (iter != candidate.end()) {
                answer.push_back(iter->second);
                answer.push_back(i);
                break;
            }
            else {
                candidate[nums[i]] = i;
            }
        }
        return answer;
    }
};