[CQOI2018]交错序列 (矩阵快速幂,数论)

[CQOI2018]交错序列



$ solution: $

这一题出得真的很好,将原本一道矩阵快速幂硬生生加入组合数的标签,还那么没有违和感,那么让人看不出来。所以做这道题必须先知道(矩阵快速幂及如何构建递推矩阵)(组合数及二项式定理)。

不知道大家有没有做过洛谷的帕秋莉手环P哥的桶,这道题中不能有相邻的两个1就是我们在构造这个交错序列时不能连续加入两个1,这个如果直接让我们求方案数(不靠虑一的个数)就是矩阵快速幂的板子了(可以自己推递推方程)。但是这1题偏偏把1的个数搭上了,我们发现1的个数是可以达到 $ 10^8 $ 级别的!所以我们不能直接把1得个数当做一个状态,这里有一个巧妙的避开1的个数的递推方程:

因为我们状态转移时,我们其实只需要知道序列末尾是1还是0,而1的个数我们不妨不管(为什么要折磨自己呢?)我们可以用 $ F[i][j][0/1] $ 为前 $ i $ 位,最后一位为 $ 0/1 $ ,满足条件的序列的1的个数的j次方和(注意是j次方和)。然后我们就可以避开1的个数(因为每一次我们如果要在序列末尾加一个1,就相当于所有长度为i的末尾为一的序列都要加上一个1,然后我们只要求 $ (y+1)^j $ 即可!这个可以用二项式定理搞一下)(这样建出来的矩阵只有90*90可以勉强过)(需要卡常)

后效性:我们避开了一的个数,但是我们需要求零得个数啊!这又怎么办呢?我们把题目要我们求的东西转化一下: $ x^ay^b=(n-y)^ay^b=sum_{i=0}^{a}C(a,i)n^i(-y)^{a-i}y^b $ 这样我们就可以用我们之前设的(满足条件的序列的1的个数的j次方和)的这一维状态求解了!

卡常:因为这一题的模数小于 $ 10^8 $ 不是 $ 10^9 $ 我们矩阵乘法的时候可以多加几次再取模(详情看代码),不然真的很难卡常数!



$ code: $

#include<iostream>
#include<cstdio>
#include<iomanip>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<ctime>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>

#define ull unsigned long long
#define ll long long
#define db double
#define inf 0x7fffffff
#define rg register int

using namespace std;

int n,a,b,m,t,f,tot;
int C[93][93],nf[95];

struct su{
    ll s[182][182];
    inline void operator *(su x){
        su y;
        for(rg i=0;i<t;++i)
            for(rg j=0;j<t;++j)
                y.s[i][j]=0;
        for(rg i=0;i<t;++i)
            for(rg j=0;j<t;++j){
                for(rg k=0;k<t;++k)
                    y.s[i][j]+=s[i][k]*x.s[k][j];
                y.s[i][j]%=m;
            }
        *this=y;
    }
}bas,ans;

inline int qr(){
    char ch;
    while((ch=getchar())<'0'||ch>'9');
    int res=ch^48;
    while((ch=getchar())>='0'&&ch<='9')
        res=res*10+(ch^48);
    return res;
}

int main(){
    //freopen(".in","r",stdin);
    //freopen(".out","w",stdout);
    n=qr(); a=qr(); b=qr(); m=qr();
    f=a+b+1; t=f<<1; nf[0]=1;
    for(rg i=0;i<f;++i){
        C[0][i]=1;
        bas.s[i][i]=1;
        bas.s[f+i][i]=1;
        bas.s[0][f+i]=1;
        nf[i+1]=(ull)nf[i]*n%m;
    }
    for(rg i=1;i<f;++i)
        for(rg j=i;j<f;++j)
            C[i][j]=(C[i][j-1]+C[i-1][j-1])%m,bas.s[i][f+j]=C[i][j];
    ans.s[0][0]=1;
    while(n){
        if(n&1)ans*bas;
        bas*bas; n>>=1;
    }
    for(rg i=0;i<=a;++i){
        rg j=a+b-i;
        rg x=(ll)C[i][a]*((a-i)&1?-1:1)*nf[i]%m;
     	rg y=(ans.s[0][j]+ans.s[0][j+f])%m;
        tot=(tot+(ll)x*y)%m;
    }printf("%d
",(tot+m)%m);
    return 0;
}

原文地址:https://www.cnblogs.com/812-xiao-wen/p/10680285.html