LOJ#3478「ROIR 2021 Day 2」好数

Description&Data Constraint

1 ≤ n ≤ 1 0 17 ， k ∈ { 0 , 1 } 。 1\le n\le 10^{17}，k\in\{0,1\}。 1≤n≤1017，k∈{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;
}