HDU 2157 How many ways??（矩阵快速幂）

  HDU 2157  How many ways??

Problem Description

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

 

 

Input

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

 

 

Output

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

Sample Input

4 40 10 21 32 320 3 20 3 33 60 11 00 22 01 22 121 2 10 1 30 0

 

 

Sample Output

2

0

1

3

【思路分析】

给你一个有向图，让你去求节点i到节点j经过k个节点的不同方案数。首先可以用一个邻接矩阵来存该图，当mat[i][j]=1时，表示节点i到节点j是连通的。求解的核心思路就是当定义一个邻接矩阵A和一个邻接矩阵B时，令B=A*A，那么B[i][j]的值就为i到j经过2个节点的不同的方案数，推出B=A^k中，B[i][j]的值就为i到j经过k个节点的不同的方案数。

代码如下：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
#define MOD 1000
int n;
struct matrix
{
    int mat[35][35];
    matrix()
    {
        memset(mat,0,sizeof(mat));
    }
};

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] += A.mat[i][k] * B.mat[k][j];
                C.mat[i][j] %= MOD;
            }
    return C;
}

matrix quickMul(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 m;
    int s,t,T;
    int a,b,k;
    //freopen("1.txt","r",stdin);
    while(scanf("%d %d",&n,&m) != EOF && (n || m))
    {
        matrix A,B;
        for(int i = 0;i < m;i++)
        {
            scanf("%d %d",&s,&t);
            A.mat[s + 1][t + 1] = 1;//表示节点(s + 1)到(t + 1)有一条路径
        }

        scanf("%d",&T);
        for(int i = 0;i < T;i++)
        {
            scanf("%d %d %d",&a,&b,&k);
            B = quickMul(A,k);
            printf("%d\n",B.mat[a + 1][b + 1]);
        }
    }
    return 0;
}