【BZOJ4204】取球游戏

最新推荐文章于 2023-08-09 21:09:01 发布

原创最新推荐文章于 2023-08-09 21:09:01 发布 · 274 阅读

0 ·

CC 4.0 BY-SA版权

【类型】做题记录同时被 3 个专栏收录

767 篇文章

订阅专栏

【OJ】BZOJ

397 篇文章

订阅专栏

【算法】动态规划

239 篇文章

订阅专栏

本文介绍了一种使用矩阵快速幂的方法来高效求解概率转移问题。具体地，通过预处理概率转移矩阵的不同幂次，进而快速计算经过特定步数后的状态概率分布。适用于起点确定、转移规则明确的概率型问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【题目链接】

点击打开链接

【思路要点】

补档博客，无题解。

【代码】

#include<bits/stdc++.h>
using namespace std;
#define MAXN	1005
#define MAXLOG	32
int n, m, K;
double f[2][MAXN], matrix[MAXLOG][MAXN][MAXN];
int main() {
	scanf("%d%d%d", &n, &m, &K);
	for (int i = 1; i <= n; i++)
		scanf("%lf", &f[0][i]);
	if (n == 1) {
		printf("%.3lf\n", f[0][1]);
		return 0;
	}
	for (int i = 1; i <= n; i++) {
		matrix[0][i][i] = (double) (m - 1) / (double) m;
		if (i != n) matrix[0][i][i + 1] = 1.0 / m;
		else matrix[0][i][1] = 1.0 / m;
	}
	for (int p = 1; p < MAXLOG; p++) {
		for (int i = 1; i <= 1; i++)
			for (int j = 1; j <= n; j++)
				for (int k = 1; k <= n; k++)
					matrix[p][i][j] += matrix[p - 1][i][k] * matrix[p - 1][k][j];
		for (int i = 2; i <= n; i++)
			for (int j = 1; j <= n; j++)
				if (j == 1) matrix[p][i][j] = matrix[p][i - 1][n];
				else matrix[p][i][j] = matrix[p][i - 1][j - 1];
	}
	for (int p = 0; p < MAXLOG; p++) {
		int now = 1 << p;
		if ((now & K) != 0) {
			K ^= now;
			for (int i = 1; i <= 1; i++)
				for (int j = 1; j <= n; j++)
					for (int k = 1; k <= n; k++)
						f[i][j] += f[i - 1][k] * matrix[p][k][j];
			for (int i = 1; i <= n; i++) {
				f[0][i] = f[1][i];
				f[1][i] = 0;
			}
		}
	}
	for (int i = 1; i <= n; i++)
		printf("%.3lf\n", f[0][i]);
	return 0;
}