豆包MarsCode 代码练习

题目

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按任意顺序返回答案。

• 1 <= nums.length <= 10^5
• k 的取值范围是 [1, 数组中不相同的元素的个数]
• 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的

你所设计算法的时间复杂度必须优于 O(n log n) ，其中 n 是数组大小。

示例 1

输入: nums = [1,1,1,2,2,3], k = 2

输出: [1,2]

示例 2

输入: nums = [1], k = 1

输出: [1]

实现的代码

#include <iostream>
#include <vector>
#include <string>
#include <unordered_map>
#include <queue>
using namespace std;
std::string solution(std::vector<int> nums, int k) {
    // Please write your code here
    // 1. 统计每个元素出现的频率
    unordered_map<int, int> frequencyMap;
    for (int num : nums) {
        frequencyMap[num]++;
    }

    // 2. 使用最小堆维护频率最高的k个元素
    priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> minHeap;
    
    for (auto& [num, freq] : frequencyMap) {
        minHeap.push({freq, num});
        if (minHeap.size() > k) {
            minHeap.pop();
        }
    }

    // 3. 输出堆中的元素
    vector<int> result;
    while (!minHeap.empty()) {
        result.push_back(minHeap.top().second);
        minHeap.pop();
    }
    string finalRes;
    for (size_t i = 0; i < result.size(); ++i) {
        finalRes += to_string(result[i]);
        if (i != result.size() - 1) {  // 如果不是最后一个元素，加上逗号或空格
            finalRes += ", ";
        }
    }
    return finalRes;
}

int main() {
    //  You can add more test cases here
    std::vector<int> nums1 = {1, 1, 1, 2, 2, 3};
    std::vector<int> nums2 = {1};
    std::cout << (solution(nums1, 2) == "1,2") << std::endl;
    std::cout << (solution(nums2, 1) == "1") << std::endl;
    return 0;
}

AI助手

以下是对您代码的检查和分析：

优点：

整体思路清晰，使用了哈希表 unordered_map 来统计元素频率，然后使用最小堆 priority_queue 来维护前 k 个高频元素，最后输出结果，逻辑较为合理。
可改进的地方：

代码的可读性可以进一步提高，比如可以添加更多的注释来解释关键步骤的作用。
在处理输出结果时，将数字转换为字符串并拼接的方式稍显繁琐。可以考虑直接使用循环输出数字，中间用逗号分隔。
总体来说，您的代码在功能上是可行的，但在一些细节方面可以进行优化，以提高代码的质量和可读性。

自己的想法

AI 对代码的分析和检查还是很有用的，能让自己写的代码的优缺点一目了然，认清自己在写代码上的不足

豆包 MarsCode - 登录
@豆包MarsCode官方