Codeforces Round #739 (Div. 3)

F2. Nearest Beautiful Number (hard version)

题目：

F2. Nearest Beautiful Number (hard version)

题意：

输入两个正整数n，k，要求找到最小的正整数m，使得m >= n，并且m中数字种数不得超过k种。

题解：

1. 如果n中数字种数如果少于等于k，则答案为n本身。
2. 1不满足。将n用字符串s表示。设有一整数p，使得s[0]…s[p - 1]数字种数等于k，s[0] ~ s[p]数字种数大于k。设s[0]…s[p - 1]中不同数字集为q，首先判断q中是否存在一个数字num，使得num尽可能小且num > s[p]（不可能存在num = s[p]）。若num存在，则使s[p + 1]，s[p + 2]…均等于q中最小值，s[p] = num即可。若num不存在，则将p倒退一位，即p–。然后重复进行s[p]++操作，若s[p] = 9，则将p再倒退一位，直到s[0]…s[p]中数字种数少于等于k。若等于k，则使s[p + 1]，s[p + 2]…均等于q（每次进行s[p]++后，需重新获取）中最小值。若少于k，则使s[p + 1]，s[p + 2]…均等于0，因为该情况下允许多加一种数字且不会破坏k得满足条件，题目要求答案尽可能小，所以置为0即可。

代码：

#include <bits/stdc++.h>
#define TT int t = read();while(t--)
#define ll long long
using namespace std;
inline ll read()
{
    ll x = 0,f = 1;
    char c = getchar();
    while(c < '0' || c > '9'){if(c == '-') f = -1;c = getchar();}
    while(c >= '0' && c <= '9') x = x * 10 + c - '0',c = getchar();
    return x * f;
}
void solve()
{
    string s;
    int k;
    cin >> s >> k;
    set<char> q;
    int p = 0;
    while(true)
    {
        if(p == s.size()) break;
        q.insert(s[p]);
        if(q.size() > k)
        {
            int ok = 0;
            q.erase(s[p]);
            for(set<char>::iterator it = q.begin();it != q.end();it++)
            {
                if(*it > s[p])
                {
                    ok = 1;
                    s[p] = *it;
                    for(int i = p + 1;i < s.size();i++) s[i] = *q.begin();
                    break;
                }
            }
            if(!ok)
            {
                p--;
                while(true)
                {
                    if(s[p] == '9') p--;
                    s[p]++;
                    q.clear();
                    for(int i = 0;i <= p;i++)
                    {
                        q.insert(s[i]);
                    }
                    if(q.size() > k) continue;
                    else
                    {
                        char num = q.size() == k ? *q.begin() : '0';
                        for(int i = p + 1;i < s.size();i++) s[i] = num;
                        break;
                    }
                }
            }
            break;
        }
        p++;
    }
    cout << s << endl;
}
int main()
{
    TT
    solve();
    return 0;
}