Problem Description
给你一个序列s,长度为n。
你需要找到一个长度为k的序列t使得它能被最多次数的从 s 中删除(可不连续且与顺序无关)。
你的任务是找到一个序列t。如果有多个可行方案,输出字典序最小的一个序列。
Input
第一行两个正整数 n 和 k ,分别表示原始序列的长度和要删除序列的长度。
第二行 n 个整数表示原始序列。
1<=k<=n<=2e5,所有数字均在int范围内。
Output
序列t,序列中的元素按照从小到大的顺序输出。如果有多个可行方案,输出字典序最小的一个序列。
Sample Input
10 4
1 3 1 3 10 3 7 7 12 3
Sample Output
1 3 3 7
Hint
对于样例:
1 3 1 3 10 3 7 7 12 3
应当输出 1 3 3 7(已加粗),第一次删除后变成 1 10 3 7 12 3
再删除一次后变为 10 12,无法再次删除。
而其他的任何长度为4的序列都只能出现一次,例如1 3 7 10受限于10的个数只能从 s 中删除一次。
#include<bits/stdc++.h>
#define N 200000+100
using namespace std;
int n,k,a[N];
int b[N],mmax = -1;//mmax用来节约一些时间
bool judge(int x)
{
int flag = 0;
for(int i = 1; i <= mmax; i++)
{
if(b[i] >= x)
{
flag += b[i]/x;//记得这里是b[i]/x
}
}
if(flag >= k)//如果大于给定的长度那就符合条件
{
return true;
}
else
{
return false;
}
}
int main()
{
int ans[N];//输出数组
int mid;
scanf("%d %d",&n,&k);
for(int i = 0; i < n; i++)
{
scanf("%d",&a[i]);
mmax = max(mmax,a[i]);//这里找一个mmax就是可以在上面函数跑的时候少跑一些
b[a[i]]++;//数据,不需要每次都跑2e5
}
int l = 1,r = n,num = 1;//num代表最大删除次数,刚开始一定要初始化为1;
while(l <= r)//二分答案
{
mid = (l+r) >> 1;
if(judge(mid))
{
l = mid+1;
num = mid;//如果这个次数能够满足条件那么我们就需要更新最大次数
}
else
{
r = mid-1;
}
}
int top = 0;
for(int i = 1; i <= N; i++)//这里直接都跑一遍就好了,这样跑出来的答案是字典序
{
//最小的,但是CF原题并没有要求这一项,所以下面那个代码在CF交上也是可以A的
for(int j = 1; j <= b[i]/num; j++)
{
ans[top++] = i;
if(top == k)//如果满足长度直接break
{
i = N;//同时打破外层循环
break;
}
}
}
for(int i = 0; i < top-1; i++)
{
printf("%d ",ans[i]);
}
printf("%d\n",ans[top-1]);
return 0;
}