hdoj 2157 How many ways??

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2157 

解题思路：矩阵快速幂

#include<iostream>
using namespace std;

int n;  //n个点,对应nxn矩阵

struct Matrix{
    int a[20][20];
}res;

Matrix multiply(Matrix x,Matrix y){
    Matrix temp;
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            temp.a[i][j]=0;
            for(int k=0;k<n;k++) temp.a[i][j]+=x.a[i][k]*y.a[k][j];
            temp.a[i][j]%=1000;
        }
    }
    return temp;
}

void calc(Matrix origin,int k){
    while(k){
        if(k&1) res=multiply(res,origin);
        k>>=1;
        origin=multiply(origin,origin);
    }
}

int main(){
    int m,t,A,B,k,i,j;
    Matrix origin;
    while(cin>>n>>m,n||m){

        //构造对应的矩阵
        for(i=0;i<n;i++){
            for(j=0;j<n;j++) origin.a[i][j]=0;
        }
        while(m--){
            cin>>i>>j;
            origin.a[i][j]=1;
        }

        cin>>t;
        while(t--){
            cin>>A>>B>>k;

            //初始化res为单位矩阵
            for(i=0;i<n;i++){
                for(j=0;j<n;j++) res.a[i][j]=0;
                res.a[i][i]=1;
            }

            calc(origin,k);
            cout<<res.a[A][B]<<endl;
        }
    }
    return 0;
}

---