Warcraft III 守望者的烦恼(利用矩阵乘法求解线性递推关系)

版权声明:本文为博主原创文章,未经博主允许不得转载。 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)内通过矩阵快速幂求出。

 

猜你喜欢

转载自blog.csdn.net/qq_35558510/article/details/90039931