2021-5-21 atcoder arc 086

Posted 2021-06-10 要坚持写博客

题目链接

题意：高桥有N个球。最初，整数Ai写在第i个球上。
他想重写一些球上的整数，以便在N个球上最多写K个不同的整数。
（即多个球上是一个同样大小的数也算一个数）
求高桥重写整数所需的最小球数。

解题思路：看数据范围我们可了解到，可以直接暴力，且只能一重循环的暴力。因为题意的最多不能超过k个不同的整数，所以我们可以用一个相当于滑动区间的想法。要求算所需重写的最小球数，从前扫描一遍，如果滑动区间内的数的数量还不足k的话，我们就往里面添加。否则我们就把区间里面最小的那个数提取出来和刚来的数作比较……。
存储的时候有一个技巧用小根堆来存储！！！

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#include<iostream>
#include<cmath>
#include<queue>
#define IOS ios::sync_with_stdio(false);cin.tie();cout.tie(0)

using namespace std;
typedef long long ll;
const int N=2e5+100;
map<string,int> q;
int cnt[N];
bool flag;
int ans=0;

const int maxn=50;
const int dx[]= {0,1,0,-1};
const int dy[]= {1,0,-1,0};
int a[maxn],b[maxn];
int a1,b1;
int n;
int path[maxn];
bool st[maxn];
int t=0;

int main()
{
    IOS;
    int n,k;
    ll ans=0;
    cin>>n>>k;
    for(int i=0;i<n;i++)
    {
        int x;
        cin>>x;
        cnt[x]++;
    }

    priority_queue<int, vector<int>, greater<int>> heap;

    for(int i=1;i<=200000;i++)
    {
        if(heap.size()<k)
        {
            heap.push(cnt[i]);
        }
        else
        {
            auto t=heap.top();
            heap.pop();
            if(t<=cnt[i])
            {
                ans+=t;
                heap.push(cnt[i]);
            }
            else{
                ans+=cnt[i];
                heap.push(t);
            }
        }
    }
    cout<<ans<<endl;
    return 0;
}