python/TangDou/CSP-J/AcWingTraining/MaiChePai_1.cpp

#include <bits/stdc++.h>
using namespace std;

const int N = 10010; // 最大数字位数
int n, k;           // n:车牌号长度，k:需要修改的数字个数
int a[N];           // 存储原始数字
int mi = 0x3f3f3f3f; // 存储最小花费，初始化为极大值
string res;         // 存储最优解的车牌号

// 结构体用于存储每个数字的信息
struct Node {
    int idx;   // 数字在原车牌中的位置
    int num;   // 原始数字
    int dist;  // 与目标数字的距离
} b[N];

int x; // 当前枚举的目标数字

// 自定义比较函数，用于排序
bool cmp(Node &a, Node &b) {
    // 优先按距离排序，距离小的排在前面
    if (a.dist != b.dist)
        return a.dist < b.dist;
    
    // 距离相同时，处理策略：
    // 如果原数字大于目标数字，选择索引小的优先修改（保证字典序小）
    // 如果原数字小于目标数字，选择索引大的优先修改（保证字典序小）
    if (a.num > x)
        return a.idx < b.idx;
    else
        return a.idx > b.idx;
}

int main() {
    // 输入处理
    cin >> n >> k;
    string s;
    cin >> s;

    // 将字符串转换为数字数组
    for (int i = 0; i < n; i++)
        a[i] = s[i] - '0';

    res = s; // 初始化最优解

    // 枚举所有可能的目标数字(0-9)
    for (x = 0; x <= 9; x++) {
        // 计算每个数字与目标数字的距离
        for (int i = 0; i < n; i++)
            b[i] = {i, a[i], abs(a[i] - x)};

        // 按照自定义规则排序
        sort(b, b + n, cmp);

        // 计算修改前k个数字的总花费
        int cost = 0;
        for (int i = 0; i < k; i++)
            cost += b[i].dist;

        // 生成临时修改后的车牌号
        string temp = s;
        for (int i = 0; i < k; i++)
            temp[b[i].idx] = x + '0';

        // 更新最优解
        if (cost < mi) { // 找到更小花费的方案
            mi = cost;
            res = temp;
        } else if (cost == mi) { // 花费相同，选择字典序更小的
            if (temp < res)
                res = temp;
        }
    }

    // 输出结果
    cout << mi << endl; // 输出最小花费
    cout << res << endl; // 输出最优车牌号
    
    return 0;
}