TJOI2017 可乐（矩阵快速幂）

Description

机器人初始在 $1$ 号城市上。机器人有三种行为：停在原地、下一个相邻的城市、自爆。它每一秒都会随机触发一种行为。现在给出图，问经过了 $t$ 秒, 机器人的行为方案数是多少？

$1<t\le 10^9,1\le N\le 30,0<M<100$

Solution

地图很小，先建出邻接矩阵。对于停在原地，让 $f_{i,i}=1$，对于自爆，可以建一个虚拟节点 $0$，然后让所有的点都向它连一条有向边，即为 $f_{i,0}=1$。

考虑 dp 的过程，设 $g_{i,j,k}$ 为从 $i$ 到 $j$ 走了 $k$ 步的方案数，可以发现 $k$ 这一维可以滚动掉。那么初始状态即为 $g_{i,j}=f_{i,j}$。运用 Floyd 的思想，转移为

$$f_{i,j}=\sum _{k=1}^n{f}_{i,k}\times f_{k,j}$$

可以发现这是矩阵乘法，所以矩阵快速幂搞定。

时间复杂度 $O(n^3\log t)$。

Code

挂个矩阵快速幂的模板

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 100 + 5, INF = 0x3f3f3f3f, mod = 1e9 + 7;
inline int read() {
	int x = 0, f = 0; char ch = 0;
	while (!isdigit(ch)) f |= ch == '-', ch = getchar();
	while (isdigit(ch)) x = (x << 3) + (x << 1) + (ch ^ 48), ch = getchar();
	return f ? -x : x;
}
int n;
struct mat {
	int m[N][N];
	mat() {
		memset(m, 0, sizeof(m));
		for (int i = 0; i < N; i++) m[i][i] = 1;
	}
};


mat mul(mat a, mat b) {
	mat c;
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= n; j++) {
			c.m[i][j] = 0; //Mention! 
			for (int k = 1; k <= n; k++) 
				c.m[i][j] += a.m[i][k] * b.m[k][j] % mod;
			c.m[i][j] %= mod;	
		}
	return c;
}
mat ksm(mat a, int k) {
	mat res;
	while (k) {
		if (k & 1) res = mul(res, a);
		k >>= 1;
		a = mul(a, a);
	}
	return res;
}
signed main() {
	n = read(); int k = read();
	
	mat a; 
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= n; j++)
			a.m[i][j] = read();
	mat ans = ksm(a, k);
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= n; j++)
			printf("%lld ", ans.m[i][j]);
		puts("");
	}		
	return 0;
}