AcWing 1083. Windy数
原题链接
中等
作者:
Anoxia_3
,
2021-05-01 01:20:41
,
所有人可见
,
阅读 353
#include <iostream>
#include <vector>
using namespace std;
const int N = 15;
int f[N][N];//f[i][j]表示:共i位数,最高位是j,且各位之间差值不小于2的数的个数
void init()
{
//1~9之间的数都是Windy
for(int i = 0 ; i < 10 ; i++) f[1][i] = 1;
for(int i = 2 ; i < N ; i++)
for(int j = 0 ; j < 10 ; j++)
for(int k = 0 ; k < 10 ; k++)
if(abs(k - j) >= 2)
f[i][j] += f[i - 1][k];
}
int dp(int n)
{
if(!n) return 0;
vector<int> nums;
while(n) nums.push_back(n % 10) , n /= 10;
int res = 0;
int last = -1;//记录上一个数字。初始时,因为最高位前面没有数字,因此为了使最高位可以任意取,所以last初始化为小于等于-1的数即可
for(int i = nums.size() - 1 ; i >= 0 ; i--)
{
int x = nums[i];
//如果i是最高位,为了避免前导零得从1开始
for(int j = (i == nums.size() - 1) ; j < x ; j++)
if(abs(j - last) >= 2)
res += f[i + 1][j];
if(abs(x - last) < 2) break; //如果与前一个数的差小于2,直接退出
last = x;
if(!i) res++;
}
//处理存在前导零的情况
for(int i = 1 ; i < nums.size() ; i++)
for(int j = 1 ; j < 10 ; j++)
res += f[i][j];
return res;
}
int main()
{
init();
int l , r;
cin >> l >> r;
cout << dp(r) - dp(l - 1) << endl;
}
%%%%%%%%%%%