数字游戏2(数位dp)

#include 
using namespace std;
const int N = 110;
int f[11][11][N];//第i位 此时去j   和取模p为多少
int  p;

int mod(int x) {
	return (x % p + p) % p;
}

void init() {
	memset(f, 0, sizeof f);
	for (int i = 0; i <= 9; i++) f[1][i][i % p]++;

	for (int i = 1; i < 11; i++) {
		for (int j = 0; j <= 9; j++) {
			for (int k = 0; k < p; k++)
				for (int x = 0; x <= 9; x++)
					f[i][j][k] += f[i - 1][x][mod(k - j)];
		}
	}
}

int dp(int n) {
	if (!n) return 1;
	vector<int>num;
	while (n) num.push_back(n % 10), n /= 10;
	int last = 0;
	int res = 0;

	for (int i = num.size() - 1; i >= 0; i--) {
		int x = num[i];
		for (int j = 0; j < x; j++) {
			res += f[i + 1][j][mod(-last)];
		}

		last += x;

		if (!i && last % p == 0) res++;
	}
	return res;
}


int main() {
	int a, b;
	while (cin >> a >> b >> p) {
		init();
		cout << dp(b) - dp(a - 1) << endl;
	}
	return 0;
}

你可能感兴趣的:(动态规划,动态规划)