Fork me on GitHub

Leetcode-初级算法-两数之和

题目

给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。
你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。

示例:

给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

自己的思路

遍历每个元素 x,并查找是否存在一个值与 target−x 相等的目标元素。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#include <stdio.h>
#include <iostream>
#include <vector>

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

复杂度分析:

  • 时间复杂度:O(n2)
    对于每个元素,我们试图通过遍历数组的其余部分来寻找它所对应的目标元素,这将耗费 O(n) 的时间。因此时间复杂度为 O(n2)。
  • 空间复杂度:O(1)

网上优秀思路

通过牺牲空间换取速度的方式。
将vector的元素添加到map的键值对中,同时检查表中是否已经存在当前元素所对应的目标元素。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#include <stdio.h>
#include <iostream>
#include <vector>
#include <map>
using namespace std;

vector<int> twoSum(vector<int>& nums, int target) {
vector<int> vecResult;
map<int, int> mapResult;
for (int i = 0; i < nums.size(); i++)
{
int complement = target - nums[i];
auto it = mapResult.find(complement);
if (it != mapResult.end())
{
vecResult.push_back(i);
vecResult.push_back(it->second);
}
mapResult[nums[i]] = i;
}
return vecResult;
}

复杂度分析:

  • 时间复杂度:O(n)
    我们只遍历了包含有n个元素的列表一次。在表中进行的每次查找只花费 O(1) 的时间。
  • 空间复杂度:O(n)
    所需的额外空间取决于哈希表中存储的元素数量,该表最多需要存储 n 个元素。

貌似使用hash_map效率会更高

本文标题:Leetcode-初级算法-两数之和

文章作者:阿西BUG

发布时间:2018年08月30日 - 19:08

最后更新:2018年09月13日 - 22:09

原始链接:http://blog.92yu.win/2018/08/30/Leetcode-初级算法-两数之和/

许可协议: 署名-非商业性使用-禁止演绎 4.0 国际 转载请保留原文链接及作者。

阿西BUG wechat
福利干货铺(YM-TALK)-- 互联网福利干货大集合
Enjoy it ? Donate for it ! 欣赏此文?求鼓励,求支持!
>