Leetcode 347. 前 K 个高频元素

1、问题分析

题目链接：https://leetcode-cn.com/problems/top-k-frequent-elements/
  本质上就是一个 map使用的问题。代码我已经进行了详细的注释，理解应该没有问题，读者可以作为参考，如果看不懂(可以多看几遍)，欢迎留言哦！我看到会解答一下。

2、问题解决

  笔者以C++方式解决。

#include "iostream"

using namespace std;

#include "algorithm"
#include "vector"
#include "queue"
#include "set"
#include "map"
#include "string"
#include "stack"


class Solution {
private:
    // 定义一个map 保存 整数数组中的元素 和 该元素出现的次数
    map<int, int> dp_map;
    // 结果数组
    vector<int> result;

    // 简化类型
    typedef pair<int, int> PAIR;

    // 值比较函数
    bool static cmp_by_value(const PAIR &lhs, const PAIR &rhs) {
        return lhs.second > rhs.second;
    }

public:
    vector<int> topKFrequent(vector<int> &nums, int k) {
        dealChen(nums, k);
        return result;
    }

    /**
     * 将整数数组，频率前 k 高的元素保存到 结果数组中
     * @param nums
     * @param k
     */
    void dealChen(vector<int> &nums, int k) {
        // 遍历整个数组
        for (int i = 0; i < nums.size(); ++i) {
            // 首先查找 map 中是否存在该元素
            auto iterator1 = dp_map.find(nums[i]);
            // 如果存在
            if (iterator1 != dp_map.end()) {
                // 取出元素的个数并 + 1
                iterator1->second += 1;
            }
            else {
                // 否则将该元素添加到map 容器中，初始化个数为 1
                dp_map.insert(make_pair(nums[i], 1));
            }
        }

        // 将 map 转化成 vector 数组
        vector<PAIR> vec(dp_map.begin(), dp_map.end());
        dp_map.clear();
        // 按照值排序 （从大到小）
        sort(vec.begin(), vec.end(), Solution::cmp_by_value);
        int i = 0;
        //选取 前 k 个元素
        for (auto iter : vec) {
            if (i >= k) {
                break;
            }
            // 保存到结果数组中
            result.push_back(iter.first);
            i++;
        }

    }
};

int main() {
    vector<int> nums = { 1, 1, 1, 2, 2, 2, 2, 3 };
    Solution *pSolution = new Solution;
    auto vector1 = pSolution->topKFrequent(nums, 2);
    for (int i = 0; i < vector1.size(); ++i) {
        cout << vector1[i] << " ";
    }
    cout << endl;
    system("pause");
    return 0;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

运行结果

有点菜，有时间再优化一下。

3、总结

  难得有时间刷一波LeetCode, 这次做一个系统的记录，等以后复习的时候可以有章可循，同时也期待各位读者给出的建议。算法真的是一个照妖镜，原来感觉自己也还行吧，但是算法分分钟教你做人。前人栽树，后人乘凉。在学习算法的过程中，看了前辈的成果，受益匪浅。
感谢各位前辈的辛勤付出，让我们少走了很多的弯路！
哪怕只有一个人从我的博客受益，我也知足了。
点个赞再走呗！欢迎留言哦！