F2. Nearest Beautiful Number (hard version)
题目:
F2. Nearest Beautiful Number (hard version)
题意:
输入两个正整数n,k,要求找到最小的正整数m,使得m >= n,并且m中数字种数不得超过k种。
题解:
- 如果n中数字种数如果少于等于k,则答案为n本身。
- 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;
}