51Nod - 1621 花钱买车牌

1621 花钱买车牌

题目来源：  CodeForces

基准时间限制：1 秒 空间限制：131072 KB 分值: 20  难度：3级算法题

一个车牌号由n位数字组成。如果一个车牌至少有k位数字是相同的，那么我们就说这个车牌漂亮的车牌。现在华沙想要改变他自己的车牌，使得他的车牌变得漂亮。当然，改车牌是要花钱的。每改变一位数字所要花费的费用等于当前位上的新旧数字之差的绝对值。那么总费用就是每位上所花费用的总和。

举例如下，

旧牌为0123，新牌为7765，那么对应第一位所花费用为|0-7|＝7，第二位为|1-7|＝6，第三位为|2-6|＝4，第四位为|3-5|＝2，总和为7+6+4+2＝19

华沙想用最少的钱，使他的车牌变得漂亮起来。现在给定n，k，和旧牌的号码，计算换牌的最少费，以及新牌的号码，

如果最少费用的号码有多个，我们取字典序最小的那个。

样例解释：

在样例中，把第二个数字换成“8”花费|9-8|＝1，把第五个数字换成“8”也花了1。

把第六个数字换成“8”花费|6-8|＝2.总费用为1+1+2＝4，新号码为“888188”

 

两个长度为n的序列比较方法如下。

存在两个序列x，y，长度都是n。

如果存在i(1≤i≤n)和任意j(1≤j＜i)使得  xi＜yi  并且  xj＝yj  ，那么我们就说x比y小。

Input

单组测试数据
第一行，两个由空格隔开的数字n和k（2≤n≤10^4，2≤k≤n），表示旧牌的位数，和至少要有k位数字相同才能构成漂亮的车牌。
第二行有n位数字，代表华沙的旧车牌。（旧车牌中只有数字）。

Output

共两行，
第一行，一个整数，代表换牌的最小费用，
第二行，n位数字，表示新的车牌。
如果最小费用的车牌有多个，输出字典序最小的那个。

Input示例

6 5
898196

Output示例

4
888188

思路：

贪心。可以先枚举key值，计算当前key值下的每个数字应该出现的次数和cost。取最小cost即其对应下的每个数字出现的次数。计算dis数组时可以枚举与key值的差值，需要优先改变比key大的数，保证字典序最小。需要注意的是如果cost相等，需要取dis数组中大数较少的dis。

输出的时候需要注意输出顺序以保证字典序最小。正向遍历时判断比key小的数，如果i数量已经够dis中存的值了，后边的i需要全部替换成key。反向遍历比key大的数。

加一组测试数据：

输入：

8 4
22294777

输出：

2
22274777

代码：

#include<stdio.h>
#include<string.h>
#include<queue>
#include<math.h>
#include<stdlib.h>
#include<algorithm>
using namespace std;
#define inf 1<<29
int n,m,dis[10],a[10005],ansdis[10],dis2[10];
int cmp(int num1[],int num2[],int key,int anskey)
{
    for(int i=0;i<n;i++)
    {
        if(a[i]>key && num1[a[i]]<dis2[a[i]]) return 1;
        if(a[i]>anskey && num2[a[i]]<dis2[a[i]]) return 0;
    }
    return 0;
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        memset(dis,0,sizeof dis);
        int k,maxx=0,cost=0,anscost=0,anskey;
        for(int i=0; i<n; i++)
        {
            scanf("%1d",&a[i]);
            dis[a[i]]++;
        }
        memcpy(dis2,dis,sizeof dis);
        for(int i=0; i<=9; i++)
        {
            if(dis[i]>maxx)
            {
                maxx=dis[i];
            }
        }
        if(maxx>=m)
        {
            printf("0\n");
            for(int i=0; i<n; i++)
                printf("%d",a[i]);
            printf("\n");
            continue;
        }
        anscost=inf;
        for(int key=0; key<=9; key++)
        {
            memcpy(dis,dis2,sizeof dis);
            maxx=dis[key];
            if(maxx==0) continue;
            cost=0;
            for(int i=1; i<=9; i++)
            {
                if(key+i<=9 && dis[key+i])
                {
                    if(m-maxx<=dis[key+i])
                    {
                        cost+=(m-maxx)*i;
                        dis[key+i]-=m-maxx;
                        maxx=m;
                        break;
                    }
                    else
                    {
                        cost+=dis[key+i]*i;
                        maxx+=dis[key+i];
                        dis[key+i]=0;
                    }
                }
                if(key-i>=0 && dis[key-i])
                {
                    if(m-maxx<=dis[key-i])
                    {
                        cost+=(m-maxx)*i;
                        dis[key-i]-=(m-maxx);
                        maxx=m;
                        break;
                    }
                    else
                    {
                        cost+=dis[key-i]*i;
                        maxx+=dis[key-i];
                        dis[key-i]=0;
                    }
                }
            }
            if(cost<anscost || (cost==anscost && cmp(dis,ansdis,key,anskey)))
            {
                anscost=cost;
                anskey=key;
                memcpy(ansdis,dis,sizeof dis);
            }

        }
        for(int i=0; i<n; i++)
        {
            if(a[i]<anskey)
            {
                if(ansdis[a[i]]==0)
                    a[i]=anskey;
                else
                    ansdis[a[i]]--;
            }
        }
        for(int i=n-1; i>=0; i--)
        {
            if(a[i]>anskey)
            {
                if(ansdis[a[i]]==0)
                    a[i]=anskey;
                else
                    ansdis[a[i]]--;
            }
        }
        printf("%d\n",anscost);
        for(int i=0; i<n; i++)
            printf("%d",a[i]);
        printf("\n");
    }
    return 0;
}