题意:高桥有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;
}