Warcraft III 守望者的烦恼

题目描述

头脑并不发达的warden最近在思考一个问题，她的闪烁技能是可以升级的，k级的闪烁技能最多可以向前移动k个监狱，一共有n个监狱要视察，她从入口进去，一路上有n个监狱，而且不会往回走，当然她并不用每个监狱都视察，但是她最后一定要到第n个监狱里去，因为监狱的出口在那里，但是她并不一定要到第1个监狱。
守望者warden现在想知道，她在拥有k级闪烁技能时视察n个监狱一共有多少种方案？

输入数据

第一行是闪烁技能的等级 k (1≤k≤10)
第二行是监狱的个数 n (1≤n≤2^31−1) 

输出数据

由于方案个数会很多，所以输出它 mod 7777777后的结果就行了

样例输入

2
4

样例输出

5

样例说明

把监狱编号1 2 3 4,闪烁技能为2级，
一共有5种方案
→1→2→3→4
→2→3→4
→2→4
→1→3→4
→1→2→4

小提示：建议用int64，否则可能会溢出

解析：动态规划+矩阵快速幂

 假设 dp[i] 表示达到监狱 i 时所有的方案数，则有 dp[i] = dp[i-1]+dp[i-2]+...+dp[i-k]（i>=k），当n的大小很大时，逐个计算肯定不行。此处借助矩阵快速幂乘法求解线性递推关系。如何将加法转换成乘法呢？假设 k = 5，a' = b * a , 则 dp[1]-->dp[n] 需要 a' = b^n * a 

 

 第一步：构造矩阵b

for(int i=1;i<k;i++) b.s[i][i+1]=1;
for(int i=1;i<=k;i++) b.s[k][i]=1;

第二步：矩阵乘法

map Matrix(map x,map y){
    map c;
    memset(c.s,0,sizeof(c.s));
    for(int i=1;i<=k;i++)
    	for(int j=1;j<=k;j++)
    		for(int kk=1;kk<=k;kk++){
    			c.s[i][j]+=x.s[i][kk]*y.s[kk][j];
				c.s[i][j]%=mod;
			}
    			
    return c;
}

第三步：矩阵快速幂

while(n){
    if(n&1) a=Matrix(a,b);
    b=Matrix(b,b);
    n>>=1;
}

 

 

完整代码：

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<cstring>
#include<algorithm>
#define ll long long
#define mod 7777777
using namespace std;
struct map{
    ll s[15][15];
}a,b;
int n,k;

map Matrix(map x,map y){
    map c;
    memset(c.s,0,sizeof(c.s));
    for(int i=1;i<=k;i++)
    	for(int j=1;j<=k;j++)
    		for(int kk=1;kk<=k;kk++){
    			c.s[i][j]+=x.s[i][kk]*y.s[kk][j];
				c.s[i][j]%=mod;
			}
    			
    return c;
}

int main(){
    cin>>k>>n;
    memset(a.s,0,sizeof(a.s));
    memset(b.s,0,sizeof(b.s));
    a.s[k][k]=1; //初始化，dp[1]=1
    for(int i=1;i<k;i++) b.s[i][i+1]=1;
    for(int i=1;i<=k;i++) b.s[k][i]=1;
    while(n){ //dp[n]需要乘以n个矩阵b
        if(n&1) a=Matrix(a,b);
        b=Matrix(b,b);
        n>>=1;
    }
    printf("%lld",a.s[k][k]);
}