HDU 6351 Beautiful Now（预处理）

题目链接：Beautiful Now

题意

给定一个没有前导零的整数 $n$，其十进制表示为 $(x_1x_2\cdots x_m)_{10}$，即 $n=\sum_{i=1}^m10^{m-i}x_i$，可以对这个整数进行 $k$ 次操作，每次操作选择两个整数 $i,j~(1\leq i\leq j\leq m,x_j\neq0)$ 并将这两个位置上的数字交换，问最终能够交换得到的最小的数字和最大的数字分别是多少。

输入

第一行为一个整数 $T~(1\leq T\leq100)$，接下去 $T$ 行每行两个整数 $n,k~(1\leq n,k\leq10^9)$。

输出

每组数据输出经过 $k$ 次合法交换后能够得到的最大值和最小值。

样例

输入
5 12 1 213 2 998244353 1 998244353 2 998244353 3
输出
12 21 123 321 298944353 998544323 238944359 998544332 233944859 998544332

题解

由于每次交换可以选择 $1\leq i\leq j\leq m$，其中 $i$ 可以等于 $j$，因此答案可以从任意小于等于 $k$ 次的合法交换中取最小值得到，题目要求在交换过程中不允许出现前导零，可以证明，如果交换的结果不含有前导零，就一定存在一种方式在不出现前导零的情况下从原数字得到结果数字。最后就是暴力枚举所有交换次数小于等于 $k$ 的全排列，每次 $O(n)$ 生成全排列会超时，可以预处理 $9$ 以内的全排列（如果长度达到 $10$ 就只有一种答案），全排列的交换次数可以用并查集找环计算。如果 $k$ 大于等于 $m-1$ 直接生成不含前导零的最小字典序和最大字典序作为答案。

过题代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <cstring>
#include <string>
#include <vector>
#include <list>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <bitset>
#include <algorithm>
#include <functional>
#include <iomanip>
#include <unordered_set>
using namespace std;

#define LL long long
const int maxn = 20;
int T, k, len;
char n[maxn], Max[maxn], Min[maxn], tmp[maxn];
int fa[maxn], per[maxn], ten[maxn];
vector<int> G[maxn][maxn];

void Init(int len) {
    for(int i = 0; i < len; ++i) {
        fa[i] = i;
    }
}

int Find(int x) {
    return x == fa[x]? x: fa[x] = Find(fa[x]);
}

void unit(int x, int y) {
    int xx = Find(x);
    int yy = Find(y);
    fa[xx] = yy;
}

int Count(int *per, int len) {
    int cnt = len;
    Init(len);
    for(int i = 0; i < len; ++i) {
        if(Find(per[i]) != Find(i)) {
            unit(per[i], i);
            --cnt;
        }
    }
    return len - cnt;
}

void Init() {
    ten[0] = 1;
    for(int i = 1; i <= 9; ++i) {
        ten[i] = ten[i - 1] * 10;
    }
    for(int len = 0; len <= 9; ++len) {
        for(int i = 0; i < len; ++i) {
            per[i] = i;
        }
        do {
            int cnt = Count(per, len);
            int num = 0;
            for(int i = 0; i < len; ++i) {
                num += per[i] * ten[i];
            }
            G[len][cnt].push_back(num);
        } while(next_permutation(per, per + len));
    }
}

int main() {
    #ifdef LOCAL
        freopen("test.txt", "r", stdin);
//    freopen("out.txt", "w", stdout);
    #endif // LOCAL
    ios::sync_with_stdio(false);

    Init();
    scanf("%d", &T);
    while(T--) {
        scanf("%s%d", n, &k);
        len = strlen(n);
        if(len == 10) {
            printf("%s %s\n", n, n);
            continue;
        }
        strcpy(Min, n);
        strcpy(Max, n);
        if(k >= len - 1) {
            sort(Min, Min + len);
            sort(Max, Max + len);
            int Index = 0;
            for(int i = 0; i < len; ++i) {
                if(Min[i] != '0') {
                    Index = i;
                    break;
                }
            }
            swap(Min[0], Min[Index]);
            for(int i = 0; i < len / 2; ++i) {
                swap(Max[i], Max[len - i - 1]);
            }
            printf("%s %s\n", Min, Max);
            continue;
        }
        for(int i = 1; i <= k; ++i) {
            int llen = G[len][i].size();
            for(int j = 0; j < llen; ++j) {
                int num = G[len][i][j];
                if(n[num % 10] == '0') {
                    continue;
                }
                for(int k = 0; k < len; ++k) {
                    tmp[k] = n[num / ten[k] % 10];
                }
                tmp[len] = '\0';
                if(strcmp(Min, tmp) > 0) {
                    strcpy(Min, tmp);
                } else if(strcmp(Max, tmp) < 0) {
                    strcpy(Max, tmp);
                }
            }
        }
        printf("%s %s\n", Min, Max);
    }

    return 0;
}