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[][];
 }p;
 int n,m;
 int t,a,b,k;
 Matrix mul(Matrix a,Matrix b)
 {
     Matrix c;
     for(int i=;i<n;i++)
     {
         for(int j=;j<n;j++)
         {
             c.mat[i][j]=;
             for(int k=;k<n;k++)
                 c.mat[i][j]=(c.mat[i][j]+a.mat[i][k]*b.mat[k][j])%;
         }
     }
     return c;
 }
 Matrix mod_pow(Matrix x,int n)
 {
     Matrix res;
     memset(res.mat,,sizeof(res.mat));
     for(int i=;i<;i++)
         res.mat[i][i]=;
     while(n)
     {
         if(n&)
             res=mul(res,x);
         x=mul(x,x);
         n>>=;
     }
     return res;
 }
 int main()
 {
     freopen("in.txt","r",stdin);
     while(cin>>n>>m)
     {
         if(n==&&m==)
             break;
         memset(p.mat,,sizeof(p.mat));
         int s,t;
         for(int i=;i<m;i++)
         {
             cin>>s>>t;
             p.mat[s][t]=;
         }
         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 ;
 }