【传递闭包】HDU 2157 How many ways??

UPD：现在才发现本题是个传递闭包

题目内容

春天到了,HDU校园里开满了花,姹紫嫣红,非常美丽. 葱头是个爱花的人,看着校花校草竞相开放,漫步校园,心情也变得舒畅. 为了多看看这迷人的校园,葱头决定,每次上课都走不同的路线去教室,但是由于时间问题,每次只能经过(k)个地方,比方说,这次葱头决定经过2个地方,那他可以先去问鼎广场看看喷泉,再去教室,也可以先到体育场跑几圈,再到教室. 他非常想知道,从A 点恰好经过(k)个点到达B点的方案数,当然这个数有可能非常大,所以你只要输出它模上1000的余数就可以了. 你能帮帮他么?? 你可决定了葱头一天能看多少校花哦。

输入格式

输入数据有多组,每组的第一行是2个整数(n,m)((0 < n le 20,m le 100)) 表示校园内共有(n)个点,为了方便起见,点从(0)到(n-1)编号,接着有(m)行,每行有两个整数$ s,t $((0le s,t<n)) 表示从(s)点能到(t)点,注意图是有向的.接着的一行是两个整数(T),表示有(T)组询问((1le Tle 100))。
接下来的(T)行,每行有三个整数$ A,B,k(,表示问你从A点到B点恰好经过)k(个点的方案数()k < 20$),可以走重复边。如果不存在这样的走法,则输出(0)。
当(n,m)都为(0)的时候输入结束

输出格式

计算每次询问的方案数,由于走法很多,输出其对1000取模的结果

样例输入

(??)
1 2
2 8

样例输出

4
()()

思路

令(Mat[i][j]=1)，表示从(i)到(j)连通，即(i,j)经过一个点到达的方案数为1。那么(∑(M[i][k] + M[k][j])),(k∈(0, N-1)), 就表示(i,j)经过两个点到达的方案数。即将矩阵(Mat^n)就可以求出(i,j)经过要求个数个点到达的方案数。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<algorithm>
using namespace std;
const int mod=1000;
int T,n,m;

struct Mat{
    int mat[40][40];
    Mat(){
	memset(mat,0,sizeof(mat));
    }
};

Mat mul(Mat A,Mat B){
    Mat 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]+=A.mat[i][k]*B.mat[k][j])%=mod;
            }
        }
    }
    return C;
}

Mat qpow(Mat A,int k) {
    Mat 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;
    while(1){
	scanf("%d%d",&n,&m);
	if(n==0&&m==0)return 0;

        Mat A,B;
        while(m--){
            scanf("%d%d",&s,&t);
            A.mat[++s][++t]=1;//因为不想从0开始循环就+1了
        }

        scanf("%d",&T);
        while (T--){
            int x,y,k;
            scanf("%d%d%d",&x,&y,&k);
            B=A;
            B=qpow(A,k);
            printf("%d
",B.mat[++x][++y]);
        }
    }
    return 0;
}