BM算法学习笔记

一种nb算法，可以求出数列的递推式。

具体过程是这样的。

我们先假设它有一个递推式，然后按位去算他的值。

for(int j=;j<now.size();++j)(delta[i]+=1ll*now[j]*f[i-j-]%mod)%=mod;

这是我们算出了f[i]应当是多少，但是f[i]有可能不是我们算出的值，所以我们记录一个delta，为我们算出的值减去f[i]的结果。

然后查看一下之前有没有出过锅。

如果出过，那么就补一个0，然后塞过去。。

if(!cnt){now.resize(i);cnt++;continue;}

cnt记录出锅次数，now记录当前递推式。

然后我们需要构造一个递推式把这一位的delta补上。

然后我们设inv为这一次的dalta除以上一次的delta。

然后我们的递推式就是在last和now之间补0，然后加一个inv，后面把所有的pre*(-inv)加进去，这样最后n这个位置会出现-delta就满足我们的要求了。

最后我们把构造递推式和当前递推式加起来。

再贪心选一个左端点最靠右的出锅递推式作为last。

正确性？？？

代码

#include<iostream>
#include<cstdio>
#include<vector>
#define N 100009
using namespace std;
typedef long long ll;
const ll mod=;
ll n,f[N],delta[N],fail[N],cnt,last;
vector<ll>cur,now,pre;
inline int rd(){
    int x=;char c=getchar();bool f=;
    while(!isdigit(c)){if(c=='-')f=;c=getchar();}
    while(isdigit(c)){x=(x<<)+(x<<)+(c^);c=getchar();}
    return f?-x:x;
}
inline ll power(ll x,ll y){
    ll ans=;
    while(y){
        if(y&)ans=ans*x%mod;x=x*x%mod;y>>=;
    }
    return ans;
}

int main(){
    n=rd();
    for(int i=;i<=n;++i)f[i]=rd();
    for(int i=;i<=n;++i){
       delta[i]=mod-f[i];
       for(int j=;j<now.size();++j)(delta[i]+=1ll*now[j]*f[i-j-]%mod)%=mod;
       if(!delta[i])continue;
       fail[cnt]=i;
       if(!cnt){now.resize(i);cnt++;continue;}
       ll inv=((mod-1ll*delta[i]*power(delta[fail[last]],mod-)%mod)%mod+mod)%mod;
       cur.clear();cur.resize(i-fail[last]-);cur.push_back(mod-inv);
       for(int j=;j<pre.size();++j)cur.push_back(1ll*pre[j]*inv%mod);
       if(now.size()>cur.size())cur.resize(now.size());
       for(int j=;j<now.size();++j)(cur[j]+=now[j])%=mod;
       if(i-now.size()>=fail[last]-pre.size())pre=now,last=cnt; //fail[last]!!!
       cnt++;now=cur;
    }
    for(int i=;i<now.size();++i)cout<<now[i]<<",";cout<<now.size();
    return ;
} 

应用：

[NOI2007]生成树计数

正解貌似是插头dp+快速幂。

然后我们发现k非常小。。。。

那么就可以对于每一个k打一个表，然后再扔到BM里跑一下，发现转移式子最大只有45。

于是就直接上矩乘。

代码

打表

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 402
using namespace std;
typedef long long ll;
ll a[N][N],n;
const int mod=;
inline ll power(ll x,ll y){
    ll ans=;
    while(y){
        if(y&)ans=ans*x%mod;x=x*x%mod;y>>=;
    }
    return ans;
}
inline ll ni(ll x){return power(x,mod-);}
inline ll matr(int n){
    for(int i=;i<=n;++i){
        for(int j=i+;j<=n;++j){
            ll x=1ll*a[j][i]*ni(a[i][i])%mod;
            for(int k=i;k<=n;++k)a[j][k]=(a[j][k]-x*a[i][k]%mod+mod)%mod;
        }
    }
    ll ans=;
    for(int i=;i<=n;++i)ans=ans*a[i][i]%mod;
    return ans;
}
int main(){
    freopen("out","w",stdout);
    int kk=;
    for(int n=;n<=;++n){
        memset(a,,sizeof(a));
        for(int i=;i<=n;++i){
          for(int k=i-;k>=&&k>=i-kk;--k)a[i][i]++,a[i][k]--;
          for(int k=i+;k<=n&&k<=i+kk;++k)a[i][i]++,a[i][k]--;
        }
        printf("%lld,",matr(n-));
    }
    return ;
}

矩阵乘法

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const int mod=;
ll top,n;
int s1[]={,};
int s2[]={,,,};
int s3[]={,,,,,,,};
int s4[]={,,,,,,,,,,,,,,,,,};
int s5[]={,,,,,,,,,,,,,
,,,,,,,,,,,,,,
,,,,,,,,,,,,,,,,,,};
int a1[]={,};
int a2[]={,,,};
int a3[]={,,,,,,,};
int a4[]={,,,,,,,,,,,,,,,,,};
int a5[]={,,,,,,,,,,,,,,,,,,
,,,,,,,,,,,,,,,,
,,,,,,,,,,,};
inline ll power(ll x,ll y){
    ll ans=;
    while(y){if(y&)ans=ans*x%mod;x=x*x%mod;y>>=;}
    return ans;
}
struct matrix{
    ll a[][];
    matrix(){memset(a,,sizeof(a));}
    matrix operator *(const matrix &b)const{
        matrix c;
        for(int i=;i<=top;++i)
          for(int j=;j<=top;++j){
              for(int k=;k<=top;++k)
              (c.a[i][j]+=a[i][k]*b.a[k][j]%mod)%=mod;
          }
        return c;
    }
}ans,Z;
inline void work1(){
    puts("");
}
inline void work2(){
    if(n<=){printf("%d\n",a2[n]);return;}
    for(int i=;i<=;++i){
        ans.a[][i]=a2[i];
        Z.a[i][]=s2[-i+];
        if(i!=)Z.a[i][i-]=;
    }
    n-=;top=;
    while(n){
        if(n&)ans=ans*Z;
        Z=Z*Z;
        n>>=;
    }
    printf("%lld",ans.a[][]);
}
inline void work3(){
    if(n<=){printf("%d\n",a3[n]);return;}
    for(int i=;i<=;++i){
        ans.a[][i]=a3[i];
        Z.a[i][]=s3[-i+];
        if(i!=)Z.a[i][i-]=;
    }
    n-=;top=;
    while(n){
        if(n&)ans=ans*Z;
        Z=Z*Z;
        n>>=;
    }
    printf("%lld",ans.a[][]);
}
inline void work4(){
    if(n<=){printf("%d\n",a4[n]);return;}
    for(int i=;i<=;++i){
        ans.a[][i]=a4[i];
        Z.a[i][]=s4[-i+];
        if(i!=)Z.a[i][i-]=;
    }
    n-=;top=;
    while(n){
        if(n&)ans=ans*Z;
        Z=Z*Z;
        n>>=;
    }
    printf("%lld",ans.a[][]);
}
inline void work5(){
    if(n<=){printf("%d\n",a5[n]);return;}
    for(int i=;i<=;++i){
        ans.a[][i]=a5[i];
        Z.a[i][]=s5[-i+];
        if(i!=)Z.a[i][i-]=;
    }
    n-=;top=;
    while(n){
        if(n&)ans=ans*Z;
        Z=Z*Z;
        n>>=;
    }
    printf("%lld",ans.a[][]);
}
int main(){
    int k;
    cin>>k>>n;
    if(k==)work1();
    else if(k==)work2();
    else if(k==)work3();
    else if(k==)work4();
    else if(k==)work5();
    return ;
}