如题,使用优先队列取每次最大的数量,-1后再压入
tk记录操作次数,>k跳出
sm为当前的数量,如果sm+tk>=t表明成为了众数
结果莫名其妙在task4和task5TLE掉两个点
提交记录
#include<iostream>
#include<queue>
using namespace std;
const int N=1e6+10;
int n,k;
int a[N],sum[N];
bool achv(int sm,priority_queue<int> q){
int tk=0;
while(1){
int t=q.top();
if(sm+tk>=t) return 1;
q.pop();
t--;
q.push(t);
tk++;
if(tk>k) return 0;
}
}
int main(){
priority_queue<int> q;
int ans=0,cnt=0,smax=0;
cin>>n>>k;
for(int i=1;i<=n;i++){
cin>>a[i];
sum[a[i]]++;
smax=max(smax,sum[a[i]]);
}
for(int i=0;i<=1e6;i++){
if(sum[i]) q.push(sum[i]);
if(sum[i]==smax) cnt++;
}
if(k==0){
cout<<cnt;
return 0;
}
if(achv(0,q)){
cout<<"pigstd";
return 0;
}
for(int i=0;i<=1e6;i++){
if(sum[i]&&achv(sum[i],q)) ans++;
}
cout<<ans;
return 0;
}
求大佬解答!qwq%%%