[BZOJ1297/Luogu4159][SCOI2009]迷路

题目链接：

BZOJ1297

Luogu4159

首先考虑距离只有(0,1)的情况

那么如果设(f[t][i][j])表示(i)到(j)走(t)时刻的方案数，有转移方程：

(f[t][i][j]=sum f[t-1][i][k]*f[t-1][j][k])

如果把(f[t])看成一个(n*n)的矩阵就是(f[t]=f[t-1]*f[t-1])

那么就可以矩阵快速幂。

现在考虑距离更大的情况：

若有一条距离为(x)的边((i,j))，那么可以把它拆成(x)条长度为(1)的边

对每一个点建一些“虚点”，(P(i,1))代表(i)，(P(i,j)(j>1))代表(i)的其他虚点，那么就将(P(i,1),P(i,2),cdots,P(i,x))连上边，再连上(P(i,x),P(j,1))即可。

矩阵忘记初始化。。调了1h+。。。

时间复杂度 (O((9n)^3log_2t))

代码：

#include <cstdio>
#include <cstring>
#define Pos(i,j) (((i)-1)*9+(j))

int n,m,t;
struct Matrix{int a[105][105];Matrix(){memset(a,0,sizeof a);}}f;

inline Matrix operator*(const Matrix &a,const Matrix &b)
{
	Matrix Res;
	for(register int i=1;i<=m;++i)
		for(register int k=1;k<=m;++k)
			for(register int j=1;j<=m;++j)
				Res.a[i][j]=(Res.a[i][j]+a.a[i][k]*b.a[k][j])%2009;
	return Res;
}

inline Matrix operator^(Matrix Bas,int x)
{
	Matrix Res;
	for(int i=1;i<=m;++i)Res.a[i][i]=1;
	for(;x;x>>=1,Bas=Bas*Bas)
		if(x&1)Res=Res*Bas;
	return Res;
}

int main()
{
	scanf("%d%d",&n,&t),m=n*9;
	for(int i=1;i<=n;++i)
	{
		for(int j=1;j<9;++j)f.a[Pos(i,j)][Pos(i,j+1)]=1;
		for(int j=1,d;j<=n;++j)
			if(scanf("%1d",&d),d)
				f.a[Pos(i,d)][Pos(j,1)]=1;
	}
	Matrix Res=f^t;
	printf("%d
",Res.a[Pos(1,1)][Pos(n,1)]);
	return 0;
}