牛客OI赛制测试赛2 B ：路径数量

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_37960603/article/details/82497664

题目传送门
k很小，就不写矩阵快速幂了
代码：

#include<bits/stdc++.h>
using namespace std;

typedef long long LL;
LL ans[50][50],rev[50][50],tmp[50][50];

int main(){

    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) scanf("%lld",&ans[i][j]);
    for(int i=1;i<=n;i++) rev[i][i]=1;
    for(int i=1;i<=k;i++){

        for(int j=1;j<=n;j++) for(int t=1;t<=n;t++)
        for(int v=1;v<=n;v++) tmp[j][t]+=rev[j][v]*ans[v][t];
        for(int j=1;j<=n;j++) for(int t=1;t<=n;t++) rev[j][t]=tmp[j][t];
        memset(tmp,0,sizeof(tmp));
    }
    printf("%d\n",rev[1][n]);
}