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;
}
posted @ 2021-10-08 07:44  尹昱钦  阅读(74)  评论(0编辑  收藏  举报