《矩阵乘法》《dp》Warcraft III 守望者的烦恼

Problem E. Warcraft III 守望者的烦恼

时间限制 1000 ms
内存限制 128 MB

题目描述

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

输入数据

第一行是闪烁技能的等级 k (1≤k≤10)k (1≤k≤10) 
第二行是监狱的个数 n (1≤n≤231−1)n (1≤n≤231−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，否则可能会溢出

解析：

迭代式：f（n）=f（n-1）+...+f(n-k)；

直接写的话，必然超内存，所以要用矩阵乘法来解决这个问题。

为了快速计算，我们也用快速幂。

AC代码：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<string>
#include<stack>
#include<queue>
#include<set>
#include<map>
#include<math.h>
#define INF 0x7FFFFFFF //int 最大值
#define modNum 7777777
#define maxSize 10

using namespace std;

int n,k;
struct Mat{
    long long mat[maxSize+5][maxSize+5];
};

Mat operator *(Mat a,Mat b)
{
    Mat c;
    // k*k矩阵乘k*k矩阵
    for(int i=0;i<k;i++)
    {
        for(int j=0;j<k;j++)
        {
            c.mat[i][j]=0;
            for(int l=0;l<k;l++)
                c.mat[i][j]=((a.mat[i][l]*b.mat[l][j])%modNum+c.mat[i][j])%modNum;
        }
    }

    return c;
}

Mat operator ^(Mat a,int n)
{
    Mat c;
    for(int i=0;i<k;i++)//初始化为单位k*k的单位阵
    {
        for(int j=0;j<k;j++)
            c.mat[i][j]=(i==j);
    }

    for(;n;n>>=1)
    {
        if(n&1) c=c*a;
        a=a*a;
        //cout<<666<<endl;
    }
    return c;
}


int main()
{
    ios::sync_with_stdio(false);
    cin>>k>>n;
    long long temp[maxSize+5];
    temp[1]=1;
    temp[0]=1;
    for(int i=2;i<=k;i++)//初始化
    {
        temp[i]=0;
        for(int j=i-1;j>=0;j--)
            temp[i]+=temp[j];
        temp[i]%=modNum;
    }


    if(n<=k)
        cout<<temp[n]<<endl;
    else
    {

        //初始化矩阵
        Mat a;
        for(int i=0;i<k;i++)
        {
            for(int j=0;j<k;j++)
            {
                if(i==0) a.mat[i][j]=1;
                else if(i==j+1) a.mat[i][j]=1;
                else a.mat[i][j]=0;
            }
        }

        Mat ans=a^(n-1);
        long long re=0;
        for(int i=0;i<k;i++)
        {
            re+=(ans.mat[k-1][i]*temp[k-i])%modNum;
            re%=modNum;
        }
        cout<<re<<endl;
    }


    return 0;
}

又掌握了一个知识点，很不错！

有一个讲矩阵乘法很不错的博客 安利一下！