牛客OI赛制测试赛2.B路径数量题解

题目

给出一个 n * n 的邻接矩阵A是一个01矩阵 .A[i][j]=1表示i号点和j号点之间有长度为1的边直接相连.求出从 1 号点 到 n 号点长度为k的路径的数目
输入描述:
第1行两个数n,k (20 ≤n ≤ 30,1 ≤ k ≤ 10)第2行至第n+1行，为一个邻接矩阵
输出描述:
题目中所求的数目
示例1
输入
4 2
0 1 1 0
1 0 0 1
1 0 0 1
0 1 1 0
输出
2

分析

floyed传递闭包直接做,实际就是求$A^k$中的第1行第n列的数,直接钜乘快速幂即可.
上代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll al[35][35],n,k,ans[35][35],bl[35][35];
int main(){
    scanf("%lld%lld",&n,&k);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            scanf("%lld",&al[i][j]);
    for(int i=1;i<=n;i++)
        ans[i][i]=1;
    while(k){
        memset(bl,0,sizeof(bl));
        if(k%2){
            for(int i=1;i<=n;i++)
                for(int j=1;j<=n;j++)
                    for(int l=1;l<=n;l++)
                        bl[i][j]=bl[i][j]+ans[i][l]*al[l][j];
            for(int i=1;i<=n;i++)
                for(int j=1;j<=n;j++)
                    ans[i][j]=bl[i][j],bl[i][j]=0;
        }
        k/=2;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                for(int l=1;l<=n;l++)
                    bl[i][j]=bl[i][j]+al[i][l]*al[l][j];
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                al[i][j]=bl[i][j],bl[i][j]=0;
    }
    printf("%lld\n",ans[1][n]);
    return 0;
}