代码随想录算法训练营day53

6 阅读2分钟

哈希表复习 1.经典两数之和,分析为什么用unordered_map 因为map可以存储key和value,而set的key就是value 插入map的时候可以采用pair的方式,也可以用数组的形式

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]); 
            if(iter != map.end()) {
                return {iter->second, i};
            }
            // 如果没找到匹配对,就把访问过的元素和下标加入到map中
            map.insert(pair<int, int>(nums[i], i)); 
        }
        return {};
    }
};

四数之和:


class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
        unordered_map<int, int> umap; //key:a+b的数值,value:a+b数值出现的次数
        // 遍历大A和大B数组,统计两个数组元素之和,和出现的次数,放到map中
        for (int a : A) {
            for (int b : B) {
                umap[a + b]++;
            }
        }
        int count = 0; // 统计a+b+c+d = 0 出现的次数
        // 再遍历大C和大D数组,找到如果 0-(c+d) 在map中出现过的话,就把map中key对应的value也就是出现次数统计出来。
        for (int c : C) {
            for (int d : D) {
                if (umap.find(0 - (c + d)) != umap.end()) {
                    count += umap[0 - (c + d)];
                }
            }
        }
        return count;
    }
};

使用set的场景 当我们遇到了要快速判断一个元素是否出现集合里的时候,就要考虑哈希法了。

所以这道题目使用哈希法,来判断这个sum是否重复出现,如果重复了就是return false, 否则一直找到sum为1为止。注意,

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;
        }
    }
};

总结:

  • 需要「有序性 / 范围查找」→ 选红黑树系列(set/map/multiset/multimap)

    • 仅存唯一值,要有序 → set
    • 存键值对,要有序 → map
    • 存重复值,要有序 → multiset
    • 存重复键的键值对,要有序 → multimap
  • 追求「极致查找效率」,无需有序 → 选哈希表系列(unordered_set/unordered_map)

    • 仅存唯一值,要高效去重 / 判重 → unordered_set
    • 存键值对,要高效查值 → unordered_map
  • 补充:自定义类型存储

    • 红黑树系列:需提供比较函数(如 < 运算符)

    • 哈希表系列:需提供哈希函数 + 相等判断函数(如 == 运算符)