牛客小白月赛13——小A的路径

题目来源:牛客网
题目链接:https://ac.nowcoder.com/acm/contest/549/E
题目类型:矩阵快速幂(传送门)

输入样例

4 5 2 1
1 2
1 3
2 3
4 1
3 4

输出样例

2

题目解析
矩阵表示第一天的时候u到v有多少条路径，然后直接做矩阵k次幂就能得到k天从u到v的路径数，最后统计一下就可以了。
AC代码

#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
int n,e,k,s;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
struct jx{
	ll jz[101][101];
	jx(){
		memset(jz,0,sizeof(jz));
	}
	jx operator*(const jx s)const{
		jx m;
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				for(int k=1;k<=n;k++){
					m.jz[i][j]=(m.jz[i][j]+jz[i][k]*s.jz[k][j])%mod;
				}
			}
		}
		return m;
	}
}my,ans;
void qpow(int k){
	while(k){
		if(k&1){
			ans=ans*my; 
		}
		my=my*my;
		k>>=1;
	}
}
int main(){
	scanf("%d%d%lld%lld",&n,&e,&k,&s);
	for(int i=1;i<=e;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		my.jz[u][v]++;
	}
	for(int i=1;i<=n;i++){
		ans.jz[i][i]=1;
	}
	qpow(k);
	ll sum=0;
	for(int i=1;i<=n;i++){
		if(i!=s) sum=sum+ans.jz[s][i],sum%=mod;
	}
	printf("%lld",sum);
	return 0;
}