一、Leetcode第202题:快乐数
思路:题目中要求检查 求和结果 是否会为 1,就两种情况,要么为1,要么无限循环。无限循环的情况就就转变为 查找 “求和结果”是否会 重复 出现,这时就要想到哈希表 unordered_set ! 直接把每次求的和存入哈希表,每次循环时在哈希表中对当前 sum 进行查找,找到就说明重复出现了!
class Solution {
public:
// 取数值各个位上的单数之和
int getSum(int n) {
int sum = 0;
while (n) {
sum += (n % 10) * (n % 10); // 获得个位数值 平方
n /= 10;
}
return sum;
}
bool isHappy(int n) {
unordered_set<int> set;
while(1) {
int sum = getSum(n);
if (sum == 1) {
return true;
}
// 如果这个sum曾经出现过,说明已经陷入了无限循环了,立刻return false
if (set.find(sum) != set.end()) {
return false;
} else {
set.insert(sum);
}
n = sum;
}
}
};
二、Leetcode第1题:两数之和
思路:可以直接用暴力解法,两个for循环,时间复杂度为O(n2),空间复杂度为O(1)。
// 暴力解法
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
for(int ii = 0; ii < nums.size(); ii++)
for(int jj = ii + 1; jj < nums.size(); jj++) {
if(nums[jj] == target - nums[ii]){
return {ii,jj};
}
}
return {};
}
};
如何降低时间复杂度呢?很明显第一个遍历for循环不能变化,第二个for循环 进行target - nums[ii] 的查找时还可以有更快的方法,那就是哈希表!(当我们需要查询一个元素是否出现过,或者一个元素是否在集合里的时候,就要第一时间想到哈希法。)
哈希表可以直接将第二个for循环的时间复杂度从O(n)降低到O(1)!
那怎么设计哈希表呢?题目要求返回值的索引,而判断的时候需要判断值的和,很明显应该使用 key-value结构!把值作为key,索引作为val,题目不要求顺序,故可以使用 unordered_map !
在for循环中,查找map中是否存在 key 与当前数组元素的值相加为target!若不存在,直接insert到map中,便于后续数组元素继续查找,然后进入下次循环;存在直接返回下标即可!
//哈希表
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
std::unordered_map <int,int> map;
for(int i = 0; i < nums.size(); i++) {
// 遍历当前元素,并在map中寻找是否有匹配的key
auto iter = map.find(target - nums[i]); // 时间复杂度 O(1)
if(iter != map.end()) {
//iter->first 为key iter->second 为value
return {iter->second, i};
}
// 如果没找到匹配对,就把访问过的元素和下标加入到map中
map.insert(pair<int, int>(nums[i], i));
}
return {};
}
};
四个重点:
- 为什么会想到用哈希表
- 哈希表为什么用map
- 本题map是用来存什么的
- map中的key和value用来存什么的