hdu 2157 How many ways?? (可达矩阵)

时间:2024-04-25 18:16:12

题意:给你一个有向图,从A 点到 B点恰好经过k个点的方案数 (k < 20), 可以走重复边

思路:利用离散数学中的可达矩阵,可达矩阵的K次幂便是从i到j走K步能到达的方案数

代码:

#include <cstdio>
#include <cstring>
#include <iostream> using namespace std; typedef long long ll;
int N,M,P;
const int MOD=;
//int MOD;
struct Matrix
{
ll m[][];
}; Matrix A;
Matrix I; Matrix multi(Matrix a,Matrix b)
{
Matrix ans;
for(int i=;i<N;i++)
{
for(int j=;j<M;j++)
{
ans.m[i][j]=;
for(int k=;k<P;k++)
{
ans.m[i][j]+=a.m[i][k]*b.m[k][j]%MOD;
}
ans.m[i][j]%=MOD;
}
}
return ans;
} Matrix power(Matrix a,int k)
{
Matrix ans=I,p=a;
while(k)
{
if(k&)
{
ans=multi(ans,p);
}
k>>=;
p=multi(p,p);
}
return ans;
} int main(int argc, char const *argv[])
{
int m;
while(cin>>N>>m)
{
if(N==&&m==) break;
M=P=N;
memset(A.m,,sizeof(A.m));
while(m--)
{
int a,b;
scanf("%d %d",&a,&b);
A.m[a][b]=;
}
for(int i=;i<N;i++)
I.m[i][i]=;
int T;
cin>>T;
while(T--)
{
int st,ed,k;
scanf("%d %d %d",&st,&ed,&k);
Matrix ans = power(A,k);
printf("%lld\n",ans.m[st][ed]);
}
}
return ;
}