How many ways??（HDU 2157）

How many ways??

Sample Input

4 4　　//n个点，m条路径
0 1　　//s->t可通
0 2
1 3
2 3
2　　//询问数
0 3 2　　//从0到3走两条路可到的方案有多少种
0 3 3
3 6
0 1
1 0
0 2
2 0
1 2
2 1
2
1 2 1
0 1 3
0 0

 

Sample Output

2

0

1

3

 

矩阵快速幂，在离散数学课例听过类似的题目，n次幂就代表n条路方案。

#include <cstring>
#include <cstdio>
#include <algorithm>
#include <iostream>
using namespace std;
typedef long long LL;
struct Matrix
{
    LL mat[25][25];
}p;
int n,m;
int t,a,b,k;
Matrix mul(Matrix a,Matrix b)
{
    Matrix c;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            c.mat[i][j]=0;
            for(int k=0;k<n;k++)
                c.mat[i][j]=(c.mat[i][j]+a.mat[i][k]*b.mat[k][j])%1000;
        }
    }
    return c;
}
Matrix mod_pow(Matrix x,int n)
{
    Matrix res;
    memset(res.mat,0,sizeof(res.mat));
    for(int i=0;i<25;i++)
        res.mat[i][i]=1;
    while(n)
    {
        if(n&1)
            res=mul(res,x);
        x=mul(x,x);
        n>>=1;
    }
    return res;
}
int main()
{
    freopen("in.txt","r",stdin);
    while(cin>>n>>m)
    {
        if(n==0&&m==0)
            break;
        memset(p.mat,0,sizeof(p.mat));
        int s,t;
        for(int i=0;i<m;i++)
        {
            cin>>s>>t;
            p.mat[s][t]=1;
        }
        cin>>t;
        while(t--)
        {
            cin>>a>>b>>k;
            Matrix ans=mod_pow(p,k);
        /*    for(int i=0;i<n;i++)
            {
                for(int j=0;j<n;j++)
                    cout<<ans.mat[i][j]<<" ";
                cout<<endl;
            }*/
            cout<<ans.mat[a][b]<<endl;
        }
    }
    return 0;
}