LeetCode 1.Two Sum

最新推荐文章于 2020-10-21 10:33:51 发布

原创最新推荐文章于 2020-10-21 10:33:51 发布 · 451 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

刷题专栏收录该内容

9 篇文章

订阅专栏

本文探讨了如何在整数数组中寻找两个数使它们的和等于特定目标值的问题。提出了两种解决方案，一种利用find和distance函数遍历查找，另一种采用map进行存储匹配。前者时间复杂度为O(n²)，后者优化至O(nlogn)。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution.

Example:
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].

分析

初步想法，默认为vector中每个元素均为正数，使用数组记录每个出现数字的位置，就类似于map，然后提交发现数据中有负数，当时的心情是：好久不刷题，都回退到这地步了吗？
改用map存储，就解决了。
发现find和distance两个神器，测试了一下，也能解决，查看其STL实现源码，居然是O（n）的！

实现代码

用时：809ms
时间复杂度： $O（n^2）$

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> result;
        //sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size(); ++i) {
            int j = target - nums[i];
            vector<int>::iterator it = find(nums.begin(), nums.end(), j);
            if (it != nums.end() && distance(nums.begin(), it) != i) {
                result.push_back(i);
                result.push_back(distance(nums.begin(), it));
                break;
            }

        }

        return result;

    }
};

用时：26ms
时间复杂度： $O（nlogn）$

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
            vector<int> result;
            map<int, int> mp;
            for (int i = 0; i < nums.size(); ++i) {
                mp.insert(pair<int, int>(nums[i], i));
            }
            for (int i =0;i < nums.size(); ++i) {
                int j = target - nums[i];
                map<int, int>::iterator it = mp.find(j);
                if (it != mp.end() && mp[j] != i) {
                    result.push_back(i);
                    result.push_back(mp[j]);
                    break;

                }
            }
            return result;

        }
};