CF118C Fancy Number(贪心,模拟)
传送门
解题思路
首先很显然可以贪心进行选择,枚举哪个数字有k个,然后优先绝对值更靠近他的数字,即可算出最小代价。
下一个要求是字典序最小。
还是先删掉绝对值靠近他的点,先从前往后正序删比他的的数字,再从后往前逆序删比他小的数字,即可保证字典序最小。
AC代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<iomanip>
#include<vector>
#include<ctime>
#include<set>
#include<queue>
#include<stack>
using namespace std;
template<class T>inline void read(T &x)
{
x=0;register char c=getchar();register bool f=0;
while(!isdigit(c))f^=c=='-',c=getchar();
while(isdigit(c))x=(x<<3)+(x<<1)+(c^48),c=getchar();
if(f)x=-x;
}
template<class T>inline void print(T x)
{
if(x<0)putchar('-'),x=-x;
if(x>9)print(x/10);
putchar('0'+x%10);
}
const int maxn=1e5+5;
string s;
int n,k,ans,anss[maxn],num[15],a[maxn],lst[15],maxr[15],maxl[15],cntl[15],cntr[15];
int cal(int x){
int res=k-num[x],l=x-1,r=x+1,ans=0;
maxl[x]=x-1;
maxr[x]=x+1;
while(l>=0||r<=9){
if(res<=0) break;
if(r<=9){
if(res<=0) break;
maxr[x]=r;
cntr[x]=min(res,num[r]);
ans+=cntr[x]*(r-x);
res-=cntr[x];
r++;
}
if(l>=0){
if(res<=0) break;
maxl[x]=l;
cntl[x]=min(res,num[l]);
ans+=cntl[x]*(x-l);
res-=cntl[x];
l--;
}
}
return ans;
}
void work(int x){
int res[maxn];
for(int i=1;i<=n;i++){
if(a[i]>=x){
if(a[i]<maxr[x]) res[i]=x;
else if(a[i]==maxr[x]&&cntr[x]){
res[i]=x;
cntr[x]--;
}else{
res[i]=a[i];
}
}
}
for(int i=n;i>=1;i--){
if(a[i]<x){
if(a[i]>maxl[x]) res[i]=x;
else if(a[i]==maxl[x]&&cntl[x]){
res[i]=x;
cntl[x]--;
}else{
res[i]=a[i];
}
}
}
for(int i=1;i<=n;i++){
if(anss[i]>res[i]){
for(int j=1;j<=n;j++){
anss[j]=res[j];
}
return;
}
if(anss[i]<res[i]) return;
}
}
int main(){
ios::sync_with_stdio(false);
memset(anss,0x3f,sizeof(anss));
cin>>n>>k>>s;
for(int i=1;i<=n;i++) a[i]=s[i-1]-'0';
for(int i=1;i<=n;i++){
num[a[i]]++;
}
ans=0x3f3f3f3f;
for(int i=0;i<=9;i++){
lst[i]=cal(i);
ans=min(ans,lst[i]);
}
for(int i=0;i<=9;i++) if(ans==lst[i]) work(i);
cout<<ans<<endl;
for(int i=1;i<=n;i++) cout<<anss[i];
cout<<endl;
return 0;
}