hdu2157 How many ways?

题目大意：给定一个有向图，问从A点恰好走k步（允许重复经过边）到达B点的方案数mod p的值把 给定的图转为邻接矩阵，即A(i,j)=1当且仅当存在一条边i->j。令C=A*A，那么C(i,j)=ΣA(i,k)*A(k,j)，实际上就 等于从点i到点j恰好经过2条边的路径数（枚举k为中转点）。类似地，C*A的第i行第j列就表示从i到j经过3条边的路径数。同理，如果要求经过k步的 路径数，我们只需要二分求出A^k即可。就是转化为矩阵，然后算矩阵的乘法。于是变成了矩阵快速幂，然后就应该不用我说了。

#include <bits/stdc++.h>
using namespace std;
#define MOD 1000
struct matrix
{
int mat[31][31];
matrix(){memset(mat,0,sizeof(mat));}
};
int n;
matrix mul(matrix A,matrix B)
{
matrix C;
for(int i = 1; i <= n; i ++)
for(int j = 1; j <= n; j ++)
for(int k = 1; k <= n; k ++)
C.mat[i][j] = (C.mat[i][j] + A.mat[i][k] * B.mat[k][j]) % MOD;
return C;
}
matrix powmul(matrix A,int k)
{
matrix B;
for(int i = 1; i <= n; i ++)
B.mat[i][i] = 1;
while(k)
{
if(k & 1) B = mul(B,A);
A = mul(A,A);
k >>= 1;
}
return B;
}
int main()
{
int s,t,m,T,a,b,k;
while(~scanf("%d%d",&n,&m)&&(n||m))
{
matrix A,B;
while(m --)
{
scanf("%d%d",&s,&t);
A.mat[s + 1][t + 1] = 1;
}
scanf("%d",&T);
while(T --)
{
scanf("%d%d%d",&a,&b,&k);
B = A;
B = powmul(A,k);
cout << B.mat[a + 1][b + 1]<<endl;
}
}
return 0;
}</span>