【HDU 4471】Homework(矩阵快速幂)

题目链接

题意

给出一个数列的前m项和一个递推关系式,但其中有另外的q(q<=100)项满足另外的一个递推式,求出第n项

Sol

q只有100

很容易想到跑到这里的时候换一下递推方式递推一次再快速幂
复杂度是$O(q maxt^3logn)$(maxt 为最大的递推阶数),会T

瓶颈?

你拿两个转移矩阵去乘是$O(maxt^3)$的
但是你拿一个向量(1*m的矩阵)去乘是$O(maxt^2)$的
于是可以考虑倍增了

预处理出转移矩阵的2的幂次方的矩阵,这样每次递推时只用乘log次转移矩阵且每次复杂度是maxt方的

然而这个题目坑点比较多,但是有很方便的解决办法

首先m可能小于maxt,并且有一些项可能需要从后面给出的递推式来算
如果我们手写判,然后暴力递推的的话,根本就打不下去啊!(是我太弱了)
这样我们要用一点巧妙的办法了
既然他们中间是一个其他的递推关系,我们不妨把它也写成一个矩阵,这样就不用暴力递推了.
然后我们把标号大的数放在向量的前面并把所有矩阵的阶数设为最大的阶数并把1给放好,
这样我们就完美避免了那些复杂的情况,项数不足的时候会自动添上 (好妙啊)

代码:

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<cstdio>
#include<queue>
#include<set>
#define Set(a,b) memset(a,b,sizeof(a))
using namespace  std;
typedef long long ll;
int n,m,q;
const int mod=1e9+7;
struct matrix{
    int a[110][110];
    inline void clear(){Set(a,0);}
    inline int* operator [](int x){return a[x];}
};
int maxt;
inline matrix Mul(matrix a,matrix b,int MAX){
    matrix c;c.clear();
    for(register int i=1;i<=MAX;++i)
        for(register int j=1;j<=maxt;++j)
            for(register int k=1;k<=maxt;++k)
                c[i][j]=(1ll*c[i][j]+1ll*a[i][k]*b[k][j]%mod)%mod;
    return c;
}
matrix f[31];
int t;
struct que{matrix c;int t;int n;inline bool operator <(que b)const{return n<b.n;}}a[110];
matrix S;
void cal(int b) {
    for (int i = 0; i < 31; ++i) {
        if (b & (1 << i)) {
            S =Mul(S,f[i],1);
        }
    }
}
void work(){
    int h=m;
    for(register int i=1;i<maxt;++i) f[0][i][i+1]=1;//要for到maxt!!!  这样就算不足maxt项通过后面的1也可以递推出来后面不足的项
    for(register int i=1;i<=30;++i) f[i]=Mul(f[i-1],f[i-1],maxt);
    for(register int i=1;i<=q;++i){
        for(register int j=1;j<maxt;++j){
            a[i].c[j][j+1]=1;
        }
    }
    for(register int i=1;i<=q;++i){
        register int nk=a[i].n;
        if(nk>n) continue;if(nk<=h) continue;
        cal(nk-h-1);
        S=Mul(S,a[i].c,1);h=nk;//
    }
    cal(n-h);
    printf("%d\n",S[1][1]);
    return;
}
void init(){
    S.clear();f[0].clear();
    for(register int i=m;i;--i) scanf("%d",&S[1][i]);
    scanf("%d",&t);maxt=max(t,m);
    for(register int i=1;i<=t;++i) scanf("%d",&f[0][i][1]);

    for(register int i=1;i<=q;++i){
        scanf("%d %d",&a[i].n,&a[i].t);
        maxt=max(maxt,a[i].t);a[i].c.clear();
        for(register int j=1;j<=a[i].t;++j) scanf("%d",&a[i].c[j][1]);//把特殊的转移也变成矩阵,这样方便
    }
    sort(a+1,a+1+q);
    return;
}
int main()
{
    int T=0;
    while(scanf("%d %d %d",&n,&m,&q)!=EOF){
        ++T;
        printf("Case %d: ",T);
        init();work();
    }
}