1560F1 - Nearest Beautiful Number (easy version)

题目：https://codeforces.com/problemset/problem/1560/F1

---

题意：

给你n和k，要求你找出>=并且是k漂亮的，k漂亮的表示x出现的不同的数不超过k。

---

题解：

当k=1时，直接遍历即可。

当k=2时，先暴力出a和b（b>a），而后进行贪心，我们只需要对满足条件的高位取b，低位取a。

即对从高到低的除了ab之外的数进行处理。

例如：12345-》13111 ；对只需要对高位进行处理，此时的a为1，b为3.

---

#include <iostream>
#include <string>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
typedef long long int ll;
const int N = 1e5 + 5;
//	memset(book, 0, sizeof(book));

string f1(string s)
{
	for (char i = '1'; i <= '9'; i++)
	{
		string ans(s.size(), i);
		if (ans >= s)
			return ans;
	}
}
string f2(string s)
{
	string ans1 = f1(s);//延续k=1的答案
	for (char a = '0'; a <= '9'; a++)
	{
		for (char b = a + 1; b <= '9'; b++)
		{
			bool flag = 1;//判断是否全部是由ab构成
			for (int i = 0; i < s.size(); i++)
			{
				if (s[i] < b)//小于最大值，开始构造
				{
					string ans = s;//改变ans为只包含2个数的字符串
					if (ans[i] < a)
						ans[i] = a;
					else ans[i] = b;
					for (int j = i + 1; j < s.size(); j++)
						ans[j] = a;
					if (ans1 > ans)//更新ans1=
						ans1 = ans;
				}
				//《---分别处理----》
				if (s[i] != a && s[i] != b)//不是全部由ab构成
				{   
					flag = 0;
					break;
				}
			}
			if (flag)return s;
		}
	}
	return ans1;
}
int main()
{
	int t;
	cin >> t;
	while (t--)
	{
		string s;
		int k;
		cin >> s >> k;
		if (k == 1)
			cout << f1(s) << endl;
		else if (k == 2)
			cout << f2(s) << endl;
	}
}