7203. 「ROIR 2021 Day 2」好数

Description&Data Constraint

\(1\le n\le 10^{17}，k\in\{0,1\}。\)

Solution

签到题

考虑暴力。

首先可以确定答案的位数和 \(x\) 是一样的。

那么看到第二类好数是包括第一类好数的，就可以先枚举出第一类好数， 再根据 \(k\) 来判断是否要构造第二类好数的第 2 种情况。

先枚举一个数字 \(i\) 表示所有数位上都是 \(i\)，判断与 \(x\) 的大小关系。

如果 \(k=1\)，再枚举一个位置 \(p\)，同时枚举一个数字 \(j\)，将全部都是 \(i\) 改为数位 \(p\) 上是 \(j\)，与 \(x\) 比较。

每次更新最小的，就是答案。

Code

#include<cstdio>
#define ull unsigned long long
using namespace std;
int k,num;
ull x,xx,ans,y,ten[20];
int main()
{
	ten[0]=1;
	for (int i=1;i<=17;++i)
		ten[i]=ten[i-1]*10;
	scanf("%llu%d",&x,&k);
	ull xx=x;
	while (xx)
	{
		++num;
		xx=(ull)(xx/10);
	}
	ans=0;
	for (int i=0;i<=9;++i)
	{
		y=0;
		for (int j=0;j<num;++j)
			y=y*10+i;
		if (y>=x&&(ans==0||ans>y)) ans=y;
		if (k)
		{
			for (int j=0;j<num;++j)
			{
				for (int u=0;u<=9;++u)
				{
					y-=(ull)i*ten[j];
					y+=(ull)u*ten[j];
					if (y>=x&&(ans==0||ans>y)) ans=y;
					y-=(ull)u*ten[j];
					y+=(ull)i*ten[j];
				}	
			}
		}
	}
	printf("%llu\n",ans);
	return 0;
}