hdu 2254 矩阵的应用

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

题意：有向图中求A点到B点路径长度为t1~t2的路径总数

离散数学中，有向图的邻接矩阵A表示所有点之间路径长度为1的路径数量，A^n则表示路径长度为n的路径数量，故需要求某两点在(A^t1)~(A^t2)的路径数量之和

View Code

 1 #include<iostream>
 2 #include<map>
 3 #include<cstring>
 4 const int N=31;
 5 const int m=2008;
 6 using namespace std;
 7 int n,len;
 8 struct Matrix{
 9     int map[N][N];
10 };
11 
12 Matrix mat[10001];
13 map<int,int>mp;
14 
15 Matrix Mul(Matrix &a,Matrix &b){
16     Matrix c;
17     for(int i=0;i<len;i++){
18         for(int j=0;j<len;j++){
19             c.map[i][j]=0;
20             for(int k=0;k<len;k++){
21                 c.map[i][j]+=a.map[i][k]*b.map[k][j];
22                 c.map[i][j]%=m;
23             }
24         }
25     }
26     return c;
27 }
28 /*
29 Matrix Pow(int n){
30     if(n==1)return mat[0];
31     else if(n&1){
32         return Mul(mat[0],Pow(n-1));
33     }else {
34         Matrix temp=Pow(n>>1);
35         return Mul(temp,temp);
36     }
37 }
38 */
39 
40 int main(){
41     while(scanf("%d",&n)!=EOF){
42         memset(mat[0].map,0,sizeof(mat[0].map));
43         mp.clear();
44         int p1,p2,k;
45         len=0;
46         while(n--){
47             scanf("%d%d",&p1,&p2);
48             if(mp.find(p1)==mp.end()){
49                 mp[p1]=len++;
50             }
51             if(mp.find(p2)==mp.end()){
52                 mp[p2]=len++;
53             }
54             mat[0].map[mp[p1]][mp[p2]]++;
55         }
56         for(int i=1;i<10001;i++){
57             mat[i]=Mul(mat[i-1],mat[0]);
58         }
59         scanf("%d",&k);
60         int v1,v2,t1,t2;
61         while(k--){
62             scanf("%d%d%d%d",&v1,&v2,&t1,&t2);
63             if(mp.find(v1)==mp.end()||mp.find(v2)==mp.end()){
64                 printf("0\n");
65                 continue;
66             }
67             int ans=0;
68             for(int i=t1-1;i<t2;i++){
69                 ans+=mat[i].map[mp[v1]][mp[v2]];
70             }
71             printf("%d\n",ans%m);
72         }
73     }
74     return 0;
75 }