AcWing 1083. 计蒜客 - T2657
原题链接
中等
作者:
_empty
,
2020-08-24 21:10:34
,
所有人可见
,
阅读 491
#include <cstdio>
#include <vector>
#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;
const int N = 15;
int f[N][10];
void init()
{
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(j - k) >= 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 = -2;
// 所有跟n位数一样,但都小于等于n的所有情况
for (int i = nums.size() - 1; i >= 0; i--)
{
int x = nums[i];
for (int j = 0; j < x; j++)
{
if (i == nums.size() - 1 && j == 0) continue;
if (abs(j - last) >= 2) res += f[i + 1][j];
}
if (abs(x - last) >= 2) last = x;
else break;
if (i == 0) res++;
}
// 所有位数小于n的位数的情况
for (int i = 1; i < nums.size(); i++)
for (int j = 1; j < 10; j++)
res += f[i][j];
return res;
}
int main()
{
int a, b;
init();
while (cin >> a >> b)
{
cout << dp(b) - dp(a - 1) << endl;
}
return 0;
}