【题解】洛谷P2233 [HNOI2002]公交车路线

题目

洛谷P2233 [HNOI2002]公交车路线

题解

矩阵优化的动态规划，等号左边的列向量中每个字母X表示经过（？）轮后，从A~X的方案数。每次右乘一个转移矩阵，表示增加一轮转车。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll MOD=1000;
int n;
struct matrix{ll s[10][10];}st,nxt;
inline matrix mul(matrix a,matrix b)
{
	matrix c; memset(c.s,0,sizeof(c.s));
	for(int i=1;i<=8;i++)
		for(int j=1;j<=8;j++)
			for(int k=1;k<=8;k++)
				c.s[i][j]=(c.s[i][j]+a.s[i][k]*b.s[k][j]%MOD)%MOD;
	return c;
}
inline matrix quick_pow(matrix a,ll b)
{
	matrix c;
	for(int i=1;i<=8;i++)for(int j=1;j<=8;j++) c.s[i][j]=i==j;
	for( ;b;b>>=1)
	{
		if(b&1) c=mul(c,a);
		a=mul(a,a);
	}
	return c;
}
inline void pre()
{
	st.s[1][1]=1;nxt.s[1][2]=nxt.s[1][8]=nxt.s[8][1]=nxt.s[8][7]=1;
	for(int i=2;i<=7;i++) nxt.s[i][i-1]=nxt.s[i][i+1]=1;
	nxt.s[6][5]=nxt.s[4][5]=0; 
}
inline void solve()
{
	pre(); nxt=quick_pow(nxt,n-1);	//第(n-1)次到D或F 
	st=mul(nxt,st); ll ans=(st.s[4][1]+st.s[6][1])%MOD;
	printf("%lld\n",ans);
}
 
int main()
{
	scanf("%d",&n);
	solve();
	
	return 0;
}