[航海协会]序列计数问题

序列计数问题

题目概述

题解

首先看到这题目，应该比较容易想到生成函数。
第$i$个数的生成函数显然是$F_i={\sum }_{j=0}^{b^i-c}{x}^j=\frac{1-x^{b^i-c+1}}{1-x}$
答案要求$\sum x_i<n$，相当于就是全部乘起来，做个前缀和取第$n-1$项，
其中，我们定义$f(S)={\sum }_{x\in S}{b}^x$表示
$$\begin{gathered} Ans=[x^n-1]\frac{{\prod }_{i=1}^{m}1-x^{b^i-c+1}}{(1-x{)}^{m+1}}=[x^n-1]\frac{{\sum }_{S\subset U}(-1{)}^{|S|}{x}^{f(S)-(c-1)|S|}}{(1-x{)}^{m+1}} \\ Ans=\sum _{S\subset U}(-1{)}^{|S|}(\genfrac{}{}{0ex}{}{n+m-1+(c-1)|S|-f(S)}{m}) \end{gathered}$$直接枚举集合$S$可以做到$O\left(2^{m}m\right)$，但都做到这里了我们何必暴力枚举集合呢？
考虑再给它化一下，
$$Ans=\frac{1}{m!}\sum _{S\subset U}(n+m-1+(c-1)|S|-f(S){)}^{\underline{m}}$$显然可以数位$dp$，我们可以将后面的下降幂看成一个多项式，然后$dp$维护。
为了同时保证后面下降幂那东西不小于$m$，我们枚举的$f(S)$显然得有一个上限，不能超过$n+m-1+(c-1)|S|$，这个上限与$|S|$有关，所以我们可以考虑先枚举$|S|$，再根据当前的上限数位$dp$。
定义$d{p}_{i,j,k,0/1}$表示在前$i$位中，选择了$j$位，并且是否达到上限的集合$S$的$f(S{)}^k$次方之和。
转移显然就是每次枚举是否加入当前的$2^i$嘛，但由于需要转移每个次方的值，我们的复杂度会达到惊人的$O\left(m^5\right)$，考虑优化。

注意到$2-b⩽c⩽b-1$，显然，当$b>2$时，有$(b^n-c+1)⩾{\sum }_{i=1}^{n-1}(b^i-c+1)$。
也就是说，我们可以尝试求出当前能够选择的字典序最大$S$，显然，比这字典序小的$S$都是可以被选择的。
这样我们就可以只对一个$n$进行数位$dp$了，优化到$O\left(m^4\right)$。
对于$b=2,c=1$时，上面的结论依旧成立。
对于$b=2,c=0$时，一种简单的方法就是先将所有小于$n$的$f(S)$全部计算，再减掉$f(S)+|S|⩾n$的部分的贡献。
但我们不妨继续思考一下我们上面的方法是否能够得到正确的答案，毕竟笔者懒得分类讨论。
我们定义$A>B$为集合$A$的字典序比集合$B$的字典序大。
如果$S^{\prime }\subset S$，由于$\forall i\in [1,\infty ),b^i-c+1>0$，所以$f(S)>f(S^{\prime })\wedge S>S^{\prime }$
所以如果我们直接枚举每个点加入是否符合$<n$的话一定可以找到满足$f(S)<n$的字典序最大集合$S$。
显然，对于$S^{\prime }<S,f(S^{\prime })<f(S)$，由于$|S^{\prime }|-|S|⩽m$，所以$f(S^{\prime })+|S^{\prime }|-f(S)-|S|⩽m$。
这也就是说上面的$n+m-1+(c-1)|S|-f(S)⩾0$。
如果$(c-1)|S|-f(S)⩾n$，也不过是一个在$[0,m)$之间数的$m$次下降幂罢了，它肯定经过$0$，贡献也自然是$0$，没啥用。
所以采用我们上面的方法是一定可以得到正确答案的。

时间复杂度$O\left(m^3\right)$。

源码

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<int,int> pii;
typedef unsigned int uint;
#define MAXN 505
#define MAXM (1<<15)+5
#define pb push_back
#define mkpr make_pair
#define fir first
#define sec second
const int lim=1000000;
const LL INF=0x3f3f3f3f3f3f3f3f;
const int mo=1e9+7;
template<typename _T>
void read(_T &x){
   _T f=1;x=0;char s=getchar();
   while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
   while('0'<=s&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}
   x*=f;
}
template<typename _T>
_T Fabs(_T x){return x<0?-x:x;}
LL gcd(LL a,LL b){return !b?a:gcd(b,a%b);}
int add(int x,int y,int p){return x+y<p?x+y:x+y-p;}
void Add(int &x,int y,int p){x=add(x,y,p);}
int qkpow(int a,int s,int p){int t=1;while(s){if(s&1)t=1ll*a*t%p;a=1ll*a*a%p;s>>=1;}return t;}
int N,m,B,c,len,a[MAXN],b[MAXN*4],lena,totb,g[55];
int pwb[55],ff[55],ans,dp[55][55][2],C[55][55],f[55];
char str[MAXN];bool mark[55];
void work(){
    for(int i=1;i<=totb;i++){
        if(b[i]<0)b[i+1]--,b[i]+=B;
        if(b[i]>=B)b[i+1]++,b[i]-=B;
    }
    while(b[totb+1])totb++;
    while(totb&&!b[totb])totb--;
}
signed main(){
    //freopen("seqcnt.in","r",stdin);
    //freopen("seqcnt.out","w",stdout);
    scanf("%s",str+1);read(m);read(B);read(c);
    len=(int)strlen(str+1);pwb[0]=1;
    for(int i=0;i<=m;i++){
        C[i][0]=C[i][i]=1;
        for(int j=1;j<i;j++)
            C[i][j]=add(C[i-1][j-1],C[i-1][j],mo);
    }
    ff[1]=1;for(int i=2;i<=m;i++)ff[i]=1ll*(mo-mo/i)*ff[mo%i]%mo;
    for(int i=1;i<=m;i++)pwb[i]=1ll*B*pwb[i-1]%mo;
    for(int i=1;i<=len;i++)N=(10ll*N+str[i]-'0')%mo;
    for(int i=len,now=1,j=1;i>0;i--){
        a[j]+=now*(str[i]-'0'),now*=10;
        lena=j;if(now==lim)now=1,j++;
    }
    while(lena){
        int now=0;
        for(int i=lena;i>0;i--){
            LL tp=1ll*lim*now+a[i];
            now=tp%B;a[i]=tp/B;
        }
        b[++totb]=now;
        while(lena&&!a[lena])lena--;
    }
    b[1]--;work();Add(N,m,mo);
    b[1]+=(c-1);work();
    if(totb>m+1)mark[m+1]=1;
    else for(int i=m;i>0;i--){
        //for(int j=1;j<=totb;j++)printf("%d ",b[j]);puts("");
        bool flag=0;for(int j=i+1;j<=totb;j++)if(b[j])flag|=1;
        if(!flag)continue;b[i+1]--;mark[i]=1,b[1]+=(c-1),work();
        if(!totb||b[totb]<0)break;
    }
    dp[m][0][!mark[m+1]]=1;
    for(int i=m;i>0;i--){
        int tp=(1ll*pwb[i]+mo-c+1)%mo;g[0]=1;
        for(int j=1;j<=m;j++)g[j]=1ll*tp*g[j-1]%mo;
        for(int j=0;j<=m;j++)for(int k=0;j+k<=m;k++)
            Add(dp[i-1][j+k][0],mo-1ll*dp[i][j][0]*g[k]%mo*C[j+k][k]%mo,mo);
        for(int j=0;j<=m;j++)Add(dp[i-1][j][0],dp[i][j][0],mo);
        if(mark[i])for(int j=0;j<=m;j++)for(int k=0;j+k<=m;k++)
            Add(dp[i-1][j+k][1],mo-1ll*dp[i][j][1]*g[k]%mo*C[j+k][k]%mo,mo);
        for(int j=0;j<=m;j++)Add(dp[i-1][j][!mark[i]],dp[i][j][1],mo),g[j]=0;
    }
    g[0]=1;
    for(int i=1;i<=m;i++)for(int j=m;j>0;j--)
        Add(g[j],1ll*add(N,mo-i,mo)*g[j-1]%mo,mo);
    for(int i=0;i<=m;i++){
        Add(f[i],dp[0][i][0],mo),
        Add(f[i],dp[0][i][1],mo);
        if(i&1)f[i]=mo-f[i];
    }
    for(int i=0;i<=m;i++)Add(ans,1ll*g[i]*f[m-i]%mo,mo);
    for(int i=1;i<=m;i++)ans=1ll*ff[i]*ans%mo;
    printf("%d\n",ans);
    return 0;
}

谢谢！！！