版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_35558510/article/details/90039931
头脑并不发达的warden最近在思考一个问题,她的闪烁技能是可以升级的,k级的闪烁技能最多可以向前移动k个监狱,一共有n个监狱要视察,她从入口进去,一路上有n个监狱,而且不会往回走,当然她并不用每个监狱都视察,但是她最后一定要到第n个监狱里去,因为监狱的出口在那里,但是她并不一定要到第1个监狱。
守望者warden现在想知道,她在拥有k级闪烁技能时视察n个监狱一共有多少种方案?
格式
输入格式
第一行是闪烁技能的等级k(1<=k<=10)
第二行是监狱的个数n(1<=n<=2^31-1)
输出格式
由于方案个数会很多,所以输出它 mod 7777777后的结果就行了
样例1
样例输入1
2 4
Copy
样例输出1
5
dp[n]指视察n个监狱的方案数,递推关系式如下
然而,n的大小达到了20亿,逐个算肯定不行。因此,我们借助矩阵快速幂乘法求解线性递推关系
k=5为例
变换矩阵 A 的右上角的(n-1)*(n-1)的小矩阵中的主对角线上填1,第n行填1。
因此dp[n]便可以通过求A^n求出。而A^n可以在log(n)内通过矩阵快速幂求出。