Leyni的车牌号 | ||||||
| ||||||
Description | ||||||
Leyni买了一辆车,这个车牌号是一个n位数(包含前导零)。他希望他的车牌号码中包含至少k个相同的数字,所以他需要花钱修改部分数字,计费规则是这样的:每更改一个数字,他将支付的费用等于原数字与新数字的差的绝对值(只是这一位数字的差的绝对值)。Leyni想知道为了能满足他,他将至少消费多少钱?得到的新车牌号又是什么(如果有多种方案都能满足他且消费最少,请提供给他字典序最小的一个新车牌号)? | ||||||
Input | ||||||
本题有多组测试数据,输入的第一行是一个整数T代表着测试数据的数量,接下来是T组测试数据。 对于每组测试数据: 第1行 包含两个以空格分隔的整数n,k (2 ≤ n ≤ 104, 2 ≤ k ≤ n)。 第2行 包含一个n位数(包含前导零)代表着Leyni的原车牌号。 | ||||||
Output | ||||||
对于每组测试数据: 第1行 输出Leyni至少要消费的金额。 第2行 输出Leyni得到的新车牌号。 | ||||||
Sample Input | ||||||
2 3 2 533 10 6 0001112223 | ||||||
Sample Output | ||||||
0 533 3 0000002223 | ||||||
Source | ||||||
哈理工2012春季校赛热身赛 2012.04.03 | ||||||
Author | ||||||
齐达拉图@HRBUST |
思路:
1、首先确定如果原序列中已经有某个数字的个数已经大于等于k了,那么直接输出0 原串即可。
2、我们再定义一个数组vis【i】表示数字i出现的次数,然后我们从0到9十个数字每一(设定为tt)个都进行一次判断,判断贪心的将其他数字改成当前数字tt需要多少花费,如果话费更少,那么再贪心字典序,将答案记录下来,如果判断出来贪心的将其他数字改成当前数字tt需要的花费和之前的最小花费相同,那么在贪心字典序,将之前的答案和当前答案判断一下,留下字典序更下的作为答案即可。
3、大概的框架已经构建出来了,那么其中的细节要如何搞定呢?我们一步一步来处理;
①首先将vis【i】统计出来
②然后从0到9枚举tt。
③然后我们再设定一个output【i】数组,表示当前贪心的方案改动数字i为数字tt的个数。那么我们再设定两个变量l,r,初始化l=tt-1,r=tt+1。然后我们第分两次贪心:
一、先贪心左边,即if(abs(l-tt)<=abs(r-tt))那么就将l这个数字改成tt,否则将r这个数字改成tt,并记录output数组。如果当前位子的数字改成了tt之后不够k,那么继续。
二、再贪心右边,即if(abs(r-tt)<=abs(l-tt))那么将r这个数字改成tt,否则将l这个数字改成tt,并记录的output数组。如果当前位子的数字改成了tt之后不够k,那么继续。
一直进行到满足了k之后,结束贪心过程,得到一个output【i】,并累加花费;
④对应得到的output【i】;我们再设定一个ans【】表示答案,对应扫两遍原串,贪心的得到ans【】:
一、先从前到后扫原串,如果output【a【i】-‘0’】>0&&a【i】-‘0’>tt那么将这个位子的字符改成tt。(我们希望变小的尽量在原串前边变小,使得字典序更小);
二、再从后向前扫原串,如果output【a【i】-‘0’】>0&&a【i】-‘0’<tt那么将这个位子的字符改成tt。(我们希望变大的尽量在原串后边变大,使得字典序更小);
处理完毕之后得到一个ans【】;
⑤如果当前贪心结果的花费小于之前贪心的结果的花费,直接将ans作为当前最优解,否则如果当前贪心的结果的花费等于之前贪心的结果的花费,那么将字典序更小的ans作为最优解。
4、最后输出最优解即可。
Ac代码:
代码写的有点挫:
#include<stdio.h>
#include<string.h>
using namespace std;
char a[100040];
char ans[100040];
char tmp[100040];
int vis[15];
int output[15];
int n,k;
int minn;
int target;
int abs(int w)
{
if(w<0)return -w;
else return w;
}
void Slove(int tt)
{
int l=tt-1;
int r=tt+1;
int cost=0;
int cont=k-vis[tt];
memset(output,0,sizeof(output));
while(1)
{
if(l<0)
{
if(vis[r]>=cont)
{
cost+=abs(r-tt)*cont;
output[r]=cont;
break;
}
else
{
cost+=abs(r-tt)*vis[r];
cont-=vis[r];
output[r]=vis[r];
r++;
}
}
else if(r>=10)
{
if(vis[l]>=cont)
{
cost+=abs(l-tt)*cont;
output[l]=cont;
break;
}
else
{
cost+=abs(l-tt)*vis[l];
cont-=vis[l];
output[l]=vis[l];
l--;
}
}
else
{
if(abs(l-tt)<=abs(r-tt))
{
if(vis[l]>=cont)
{
cost+=abs(l-tt)*cont;
output[l]=cont;
break;
}
else
{
cost+=abs(l-tt)*vis[l];
cont-=vis[l];
output[l]=vis[l];
l--;
}
}
else
{
if(vis[r]>=cont)
{
cost+=abs(r-tt)*cont;
output[r]=cont;
break;
}
else
{
cost+=abs(r-tt)*vis[r];
cont-=vis[r];
output[r]=vis[r];
r++;
}
}
}
}
if(cost<minn)
{
minn=cost;
for(int i=0;i<n;i++)
{
if(a[i]-'0'>tt&&output[a[i]-'0']>0)
{
ans[i]=tt+'0';
output[a[i]-'0']--;
}
else ans[i]=a[i];
}
for(int i=n-1;i>=0;i--)
{
if(a[i]-'0'<tt&&output[a[i]-'0']>0)
{
ans[i]=tt+'0';
output[a[i]-'0']--;
}
}
ans[n]='\0';
}
else if(minn==cost)
{
for(int i=0;i<n;i++)
{
if(a[i]-'0'>tt&&output[a[i]-'0']>0)
{
tmp[i]=tt+'0';
output[a[i]-'0']--;
}
else tmp[i]=a[i];
}
for(int i=n-1;i>=0;i--)
{
if(tmp[i]-'0'<tt&&output[a[i]-'0']>0)
{
tmp[i]=tt+'0';
output[a[i]-'0']--;
}
}
if(strcmp(tmp,ans)<0)
{
for(int i=0;i<n;i++)ans[i]=tmp[i];
}
}
}
void Slove2(int tt)
{
int l=tt-1;
int r=tt+1;
int cost=0;
int cont=k-vis[tt];
memset(output,0,sizeof(output));
while(1)
{
if(l<0)
{
if(vis[r]>=cont)
{
cost+=abs(r-tt)*cont;
output[r]=cont;
break;
}
else
{
cost+=abs(r-tt)*vis[r];
cont-=vis[r];
output[r]=vis[r];
r++;
}
}
else if(r>=10)
{
if(vis[l]>=cont)
{
cost+=abs(l-tt)*cont;
output[l]=cont;
break;
}
else
{
cost+=abs(l-tt)*vis[l];
cont-=vis[l];
output[l]=vis[l];
l--;
}
}
else
{
if(abs(r-tt)<=abs(l-tt))
{
if(vis[r]>=cont)
{
cost+=abs(r-tt)*cont;
output[r]=cont;
break;
}
else
{
cost+=abs(r-tt)*vis[r];
cont-=vis[r];
output[r]=vis[r];
r++;
}
}
else
{
if(vis[l]>=cont)
{
cost+=abs(l-tt)*cont;
output[l]=cont;
break;
}
else
{
cost+=abs(l-tt)*vis[l];
cont-=vis[l];
output[l]=vis[l];
l--;
}
}
}
}
if(cost<minn)
{
minn=cost;
for(int i=0;i<n;i++)
{
if(a[i]-'0'>tt&&output[a[i]-'0']>0)
{
ans[i]=tt+'0';
output[a[i]-'0']--;
}
else ans[i]=a[i];
}
for(int i=n-1;i>=0;i--)
{
if(a[i]-'0'<tt&&output[a[i]-'0']>0)
{
ans[i]=tt+'0';
output[a[i]-'0']--;
}
}
ans[n]='\0';
}
else if(minn==cost)
{
for(int i=0;i<n;i++)
{
if(a[i]-'0'>tt&&output[a[i]-'0']>0)
{
tmp[i]=tt+'0';
output[a[i]-'0']--;
}
else tmp[i]=a[i];
}
for(int i=n-1;i>=0;i--)
{
if(tmp[i]-'0'<tt&&output[a[i]-'0']>0)
{
tmp[i]=tt+'0';
output[a[i]-'0']--;
}
}
if(strcmp(tmp,ans)<0)
{
for(int i=0;i<n;i++)ans[i]=tmp[i];
}
}
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
memset(vis,0,sizeof(vis));
scanf("%d%d",&n,&k);
scanf("%s",a);
int flag=0;
target=0;
minn=0x3f3f3f3f;
for(int i=0;i<n;i++)
{
vis[a[i]-'0']++;
if(vis[a[i]-'0']>=k)flag=1;
}
if(flag==1)
{
printf("0\n");
printf("%s\n",a);
}
else
{
for(int i=0;i<10;i++)
{
Slove(i);
Slove2(i);
}
printf("%d\n",minn);
printf("%s\n",ans);
}
}
}