AcWing 1084. 个人理解,只需要预处理一次
原题链接
中等
作者:
SmallBee
,
2024-01-17 14:57:59
,
所有人可见
,
阅读 48
#include <iostream>
#include <vector>
#include <cstring>
using namespace std;
const int N = 35, M = 320;
int l, r, p;
int f[N][M]; //f[i][j]表示前i位总和为j的方案数
void init()
{
//由于第二维记录的是总和而不是取模的值,所以只用预处理一次
f[0][0] = 1;
for (int i = 1; i <= N; i++)
for (int j = 0; j <= i * 9; j++)
for (int k = 0; k <= 9; k++)
if (j >= k)
f[i][j] += f[i - 1][j - k];
}
int dp(int n)
{
if (!n) return 1;
vector<int> num;
while (n)
{
num.push_back(n % 10);
n /= 10;
}
int res = 0, last = 0;
for (int i = num.size() - 1; i >= 0; i--)
{
int x = num[i];
for (int j = 0; j < x; j++) //枚举这一位选0 ~ x-1
for (int k = 0; k <= i * 9; k++) //枚举下一位的总和
if ((j + k + last) % p == 0) //取模操作放在这里
res += f[i][k];
last += x;
if (!i && last % p == 0) res++;
}
return res;
}
int main()
{
init();
while(cin >> l >> r >> p)
cout << dp(r) - dp(l - 1) << endl;
return 0;
}
这9104
Orz
Orz