洛谷 P7158 的题解

比较有意思的一题。

题目大意

求没有前导 $0$ 且有偶数个 $k$ 的 $n$ 位数的数量。（题意很简单是不是？）

思路

核心算法： 动态规划

我们设 $o{k}_i$ 是满足题目要求的 $i$ 位数，$n{o}_i$ 是不满足题目要求的 $i$ 位数。

橙题的转移都是很简单的，我们慢慢寻找两个数组的关系。
（请读者仔细思考）

经过简单的分析，可以得出 $o{k}_i=(o{k}_{i-1}\times 9+n{o}_{i-1})$，$n{o}_i=(n{o}_{i-1}\times 9+o{k}_{i-1})$，正好是反着的。

这样 dp 就被我们如此简单的解决了。

初始化挺好写，见代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define mod 998244353
int T, n, k;
ll ok[100007], no[100007];
inline int read()
{
	int x = 0, f = 1;
	char ch = getchar();
	while(ch < '0' || ch > '9')
	{
		if(ch == '-') f = -1;
		ch = getchar();
	}
	while(ch >= '0' && ch <= '9')
	{
		x = (x << 1) + (x << 3) + (ch ^ 48);
		ch = getchar();
	}
	return x * f;
}
int main()
{
	T = read();
	ok[1] = 8; no[1] = 1;
	for(int i = 2; i <= 100000; i++)
	{
		ok[i] = (ok[i - 1] * 9 + no[i - 1]) % mod;
		no[i] = (no[i - 1] * 9 + ok[i - 1]) % mod;
	}
	while(T--)
	{
		n = read(), k = read();
		if(n == 1) { printf("9\n"); continue; }
		printf("%ld\n", ok[n]);
	}
	return 0;
}