路径数量

链接：https://www.nowcoder.com/acm/contest/185/B
来源：牛客网
 

题目描述

给出一个 n * n 的邻接矩阵A.

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

说明

样例如图:

第一条路径：1-2-4

第二条路径：1-3-4

 

裸的弗洛伊德 传递闭包， 矩阵快速幂一下就好。 其实直接矩阵乘就能过

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<ctime>
#include<cstring>
#define int long long
using namespace std;
int n,k;
struct _Martix{
int
num[210][210];
_Martix(){memset(num,0,sizeof(num));}
friend _Martix operator*
( const _Martix &__a,const _Martix &__b)
{
_Martix __c;
int _n=n;
for( int _k = 1 ; _k <= _n ; _k ++ )
for( int _i = 1 ; _i <= _n ; _i ++ )
for( int _j = 1; _j <= _n ; _j ++ )
__c.num[_i][_j] +=
__a.num[_i][_k] * __b.num[_k][_j] ;
return __c;
}
}q;
_Martix pow2(_Martix a,long long b)
{
_Martix c;
for(int i=1;i<=n;i++) c.num[i][i]=1;
for(;b;b>>=1,a=a*a) if(b&1) c=c*a;
return c;
}
signed main()
{
scanf("%lld%lld",&n,&k);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
scanf("%lld",&q.num[i][j]);
q=pow2(q,k);
printf("%lld",q.num[1][n]);
return 0;
}