hdu 2157 How many ways??

How many ways??

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 369    Accepted Submission(s): 97

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 4
0 1
0 2
1 3
2 3
2
0 3 2
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

 

Author

小黑

 

Source

2008信息工程学院集训队——选拔赛

 

Recommend

linle

 

Statistic | Submit | Back

//1345423 2009-05-07 14:01:37 Accepted 2157 15MS 248K 1068 B C++ Wpl 
#include <iostream>
#define MAX 22
using namespace std;
int dp[MAX][MAX][MAX];
bool mark[MAX][MAX];
int n,m,t;
void Init()
{
    int i,j,a,b,k;
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
            mark[i][j]=false;
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
            for(k=0;k<=20;k++)
            {
                if(i==j&&k==0)  //初始化i到i经过0步有一种方法
                {
                    dp[i][j][k]=1;
                    continue;
                }
                dp[i][j][k]=0;
            }
    for(i=0;i<m;i++)
    {
        scanf("%d%d",&a,&b);
        mark[a][b]=true;
        dp[a][b][1]=1;
    }
}
void Floyd()
{
    int i,j,k,p;
    for(p=2;p<=20;p++)
        for(k=0;k<n;k++)
            for(i=0;i<n;i++)
                for(j=0;j<n;j++)
                {
                    if(mark[k][j])
                    {
                        dp[i][j][p]=(dp[i][j][p]+dp[i][k][p-1])%1000;  //对1000取模
                    }
                }
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(n==0&&m==0)
            break;
        Init();
        Floyd();
        scanf("%d",&t);
        int a,b,k;
        while(t--)
        {
            scanf("%d%d%d",&a,&b,&k);
            /**//*if(k==0&&a==b)
            {
                printf("1\n");
                continue;
            }
            if(k==0&&a!=b)
            {
                printf("0\n");
                continue;
            }*/
            printf("%d\n",dp[a][b][k]);
        }
    }
    return 0;
}

//1357418 2009-05-11 13:04:36 Time Limit Exceeded 2157 1000MS 196K 1600 B C++ Wpl 
//1358305 2009-05-11 19:12:02 Accepted 2157 15MS 260K 1818 B C++ Wpl 
#include <iostream>
#define MAX 22
using namespace std;
typedef struct node
{
    int matrix[MAX][MAX];
}Matrix;
Matrix unit,init,dp[MAX];
int n,m,t;
void Init()
{
    int i,j,a,b;
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
        {
            init.matrix[i][j]=0;  //对角线初始化为0,默认i到i是没有路的
            unit.matrix[i][j]=(i==j);
        }
    for(i=0;i<m;i++)
    {
        scanf("%d%d",&a,&b);
        init.matrix[a][b]=1;
    }
}
Matrix Mul(Matrix a,Matrix b)//据说传结构体比传数组快
{
    int i,j,k;
    Matrix c;
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
        {
            c.matrix[i][j] = 0;
            for(k=0;k<n;k++)
                c.matrix[i][j] += a.matrix[i][k]*b.matrix[k][j];
            c.matrix[i][j]%=1000;
        }
    return c;
}
Matrix Cal(int exp)//exp代表幂,这里是利用二分法求矩阵的幂
{
    Matrix p,q;
    p = unit;   //p为单位矩阵
    q = init;   //q为初始矩阵
    while(exp!=1)
    {
        if(exp&1)  //exp是奇数,实际这里只进行一次运算
        {
            exp--;
            p = Mul(p,q);  //如果k是奇数,那么就不能进行平均的二分,所以让p乘以一个单位矩阵,保证其不变,然后k--就可以进行二分了
        }
        else  //exp是偶数
        {
            exp>>=1;  //exp除2
            q = Mul(q,q);
        }
    }
    p = Mul(p,q);//如果k是偶数的话这一步是没有必要的
    return p;
}
int main()
{
    int s,e,k,i;
    Matrix c;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(n==0&&m==0)  //一开始忘记加这个所以超时
            break;
        Init();
        for(i=1;i<20;i++)
            dp[i]=Cal(i);
        scanf("%d",&t);
        while(t--)
        {
            scanf("%d%d%d",&s,&e,&k);
            if(s==e&&k==0)
            {
                printf("1\n");
                continue;
            }
            printf("%d\n",dp[k].matrix[s][e]);
        }
    }
    return 0;
}