[补题记录] Atcoder Beginner Contest 300（E）

URL：https://atcoder.jp/contests/abc300

目录

E

Problem/题意

Thought/思路

Code/代码

---

E

Problem/题意

给一个 N，然后投色子，筛到每个数（即 1 - 6）的概率均等，把所筛到的数一个个相乘，直到 T = N 的概率是多少。

Thought/思路

来自：https://blog.csdn.net/weixin_66946161/article/details/130464789

dp[x] 表示，从 x 到 N 的概率是多少。

Code/代码

#include "bits/stdc++.h"
#define int long long 

const int mod = 998244353;

int ksm(int a, int b) {
	int res = 1;
	while (b > 0) {
		if (b & 1) res = res * a % mod;
		a = a * a % mod;
		b = b / 2;
	}
	return res;
}

int inv(int x, int p) {
	return ksm(x, p - 2);
}

int n, inv5;
std::map  dp;

int dfs(int x) { // dp[x]: x 到 n 的概率
	if (x > n) return 0; // 大于 n 的情况不存在
	else if (x == n) return 1; // n 到 n 的概率为 1
	else if (dp.count(x)) return dp[x];

	int res = 0;
	for (int i = 2; i <= 6; ++ i) res += dfs(i * x);

	res = res * inv5 % mod;
	dp[x] = res;

	return dp[x];
}

signed main() {
	std::cin >> n;

	inv5 = inv(5, mod);

	std::cout << dfs(1);
}