PATA1129_用vector也可以做

最新推荐文章于 2021-08-23 11:44:19 发布

原创最新推荐文章于 2021-08-23 11:44:19 发布 · 104 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

PAT甲级刷题专栏收录该内容

124 篇文章

订阅专栏

这两篇博客分别用C++实现了在给定限制时间内动态维护数组中K个最大元素的方法。第一篇通过自定义结构体和排序优化了时间效率，第二篇利用了集合（set）进行元素管理和比较，确保了K个最大元素的实时更新。两篇代码都着重于性能优化和内存管理。

一开始自己尝试vector超时，然后想了一下剪掉一些没必要的排序。题目要求是400ms，其实还好

//8:10 ~ 
#include<bits/stdc++.h>
using namespace std;
struct node {
    int val, times;
}Node[50010];
vector<int> v;
bool vis[50010];
bool cmp (int a, int b) {
    if (Node[a].times != Node[b].times) return Node[a].times > Node[b].times;
    return a < b;
}
int main() {
    int n, k, pre, now;
    scanf ("%d %d %d", &n, &k, &pre);
    vis[pre] = true;
    v.push_back(pre);
    Node[pre].times++;
    for (int i = 1; i < n; i++) {
        scanf ("%d", &now);
        printf ("%d:", now);
        for (int j = 0; j < v.size(); j++) {
            printf (" %d", v[j]);
        }
        printf ("\n");
        Node[now].times++;
        if (vis[now] == false) {
            if (v.size() == k) {
                if (Node[now].times > Node[v[k - 1]].times) {
                    vis[v[k - 1]] = false;
                    v[k - 1] = now;
                    vis[now] = true;
                }
                else if (Node[now].times == Node[v[k - 1]].times && now < v[k - 1]) {
                    vis[v[k - 1]] = false;
                    v[k - 1] = now;
                    vis[now] = true;
                }
            }
            else {
                v.push_back(now);
                vis[now] = true;
            }
        }
        sort (v.begin(), v.end(), cmp);
    }
}

重载运算符

#include <bits/stdc++.h>
#include <set>
using namespace std;
int vis[50010];
struct node {
    int val, cnt;
    node(int a, int b):val(a), cnt(b){};
    bool operator < (const node &a) const {
        return a.cnt != cnt ? cnt > a.cnt : val < a.val;
    }
};
set<node> st;
int main() {
    int n, k, num;
    scanf ("%d %d %d", &n, &k, &num);
    vis[num]++;
    st.insert(node(num, vis[num]));
    for (int i = 1; i < n; i++) {
        int tempcnt = 0;
        scanf ("%d", &num);
        printf ("%d:", num);
        for (auto it = st.begin(); it != st.end() && tempcnt < k; it++, tempcnt++) {
            printf (" %d", it->val);
        }
        printf ("\n");
        auto it = st.find(node(num, vis[num]));
        if (it != st.end()) st.erase(it);
        vis[num]++;
        st.insert(node(num, vis[num]));
    }
}