题目链接:https://leetcode.com/problems/two-sum/
简述:
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, and you may not use the same element twice.
Example:
Given nums = [2, 7, 11, 15], target = 9, Because nums[0] + nums[1] = 2 + 7 = 9, return [0, 1].
题目很简单,给定一串数组,和一个目标值,要你从数组中找出两个数相加等于目标值。
首先给出最优解,时间复杂度为O(n):
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
//Key is the number and value is its index in the vector.
unordered_map<int, int> hash;
vector<int> result;
for (int i = 0; i < nums.size(); i++) {
int numberToFind = target - nums[i];
//if numberToFind is found in map, return them
if (hash.find(numberToFind) != hash.end()) {
//+1 because indices are NOT zero based
result.push_back(hash[numberToFind] );
result.push_back(i );
return result;
}
//number was not found. Put it in the map.
hash[nums[i]] = i;
}
return result;
}
};
(ps:这代码不是我想出来的,代码来源:https://leetcode.com/problems/two-sum/discuss/13/Accepted-C%2B%2B-O(n)-Solution)
然后分享我的解题思路:
很自然的用暴力方法过了一遍,复杂度O():
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
for(int i=0;i<nums.size()-1;i++)
for (int j = i + 1; j < nums.size(); j++)
if (nums[i] + nums[j] == target)
return {i,j};
return {};
}
};
然后发现70%的人时间比我快。所以推测这个肯定有更好的办法。花了一分钟想了一下没有思路,打开讨论看别人的代码,看了热度最高的两个(上面最优解)
思路:,遍历数组a,计算a[i]与target的差值,把差值作为key在map里查找value,若找到则输出{i,value},否则把差值与下标i作为key-value存入map。
数组遍历一遍,map遍历一遍,不还是O()么?于是查了一下最优解里unordered_map里的find函数,unordered_map存储是哈希所以find复杂度是O(1),完毕;
优先使用map(或者unordered_map)的find函数而非algorithm里的find函数: https://www.cnblogs.com/daocaorenblog/p/5685412.html
其他问题:
CE:复制代码的时候少了花括号。
WA:target为0的时候没有返回。
AC1:爆力O();
AC2:优化了下O(),滑稽。
AC3:最优解代码。
以前写过一点编程题都是int main(),这里的用函数封装。
vector现学的。
unordered_map现学的。
哈希是学过的,但是刷题又不会手写个哈希,以至于哈希思想也不会想到,所以熟练运用ST很重要。