求助task4 task5莫名T1个点
查看原帖
求助task4 task5莫名T1个点
318225
pur_lin楼主2023/7/15 18:15

如题,使用优先队列取每次最大的数量,-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%%%

2023/7/15 18:15
加载中...