目录
引言
工作久了之后,突然发现大学时候的算法题离自己越来越远了,刷再多的题,也只是在面试的时候用到。也不知怎么的,最近特别想刷题。为了坚持下去,每日在CSDN博客打卡。
题目
(原题链接:https://leetcode-cn.com/problems/two-sum/)
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9。
因为 nums[0] + nums[1] = 2 + 7 = 9,
所以返回 [0, 1]。
解题
方法一、纯暴力
因为题目中要求同一元素不能使用两次,所以可以采用每个元素和后面每个元素相加求和再与目标值比较的方法。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> res;
for (int i = 0; i < nums.size(); i ++) {
for (int j = i+1; j < nums.size(); j++) {
if (nums[i] + nums[j] == target) {
res.push_back(i);
res.push_back(j);
}
}
}
return res;
}
};
执行结果:
方法二、哈希表法
因为题目结果只对应一个答案,所以可以认为数组中没有相同元素,简化模型。这样我们可以构建一个哈希表,键是数组元素的值,值是数组元素的下标。这里可能跟我们脑海中的模型是反着的,不能理解也没有关系,直接看代码就懂了:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> res;
unordered_map<int,int> map; // 这里本想用hash_map的,但是它还没有成为C++11的正式标准
for (int i = 0; i < nums.size(); i ++) {
if (map.find(target-nums[i]) != map.end()) {
res.push_back(map[target-nums[i]]);
res.push_back(i);
}
map[nums[i]] = i; // 放在if的后边,是为了避免i=0时的误判
}
return res;
}
};
执行结果: