NYG的序列拆分

2021-09-15 20:34:20 阅读：180 来源： 互联网

标签：lfloor log leq 公比拆分序列 include NYG dp

Description

问长为 \(n\) ，每个数值域在 \([l,r]\) 的所有 \((r-l+1)^n\) 个序列的每个序列的拆分的方式之和。拆分合法当且仅当，将序列分为若干段，每一段都是一个等比数列。\(1\leq l\leq r\leq 10^7\)，\(n\leq 10^{18}\)。

Solution

容易发现问题可以拆解为求不同长度的等比数列个数和将这些数列组合起来。

先抛开公比是一的情况。那么对于长为 \(n\) ，公比为 \(\frac{x}{y}\) 的数列，一定长成

\[ty^{n-1}，ty^{n-2}x，\dots，tx^{n-1} \]

的样子。\(n=1\) 和 \(n=2\) 比较特殊。单独考虑，容易得到 \(cnt_{1}=r-l+1\) 和 \(cnt_2=(r-l+1)\times(r-l)\)。对于其它情况，\(n\geq 3\)，那么有 \(\max(x^2,y^2)\leq r\)，所以只用在 \(\sqrt r\) 下枚举。容易发现这样的数列长度最大为 \(\min(\log_x n,\log_y n)\)。那么如果枚举长度，就可以快速统计该种长度的数列个数 \(\max(0,\lfloor \frac{r}{x^{n-1}} \rfloor -\lfloor \frac{l-1}{y^{n-1}} \rfloor)\)。

如果题目保证公比不为一，那么就可以直接矩阵快速算答案了。因为有转移

\[dp_i=\sum_{j=1}^{\lfloor \log r \rfloor} dp_{i-j}\times f_j \]

\(f\) 表示等比数列个数。

现在加上公比是一的情况，有

\[dp_i=\sum_{j=1}^{\lfloor \log r \rfloor} dp_{i-j}\times f_j+(r-l+1)\times\sum_{j=2}^{i} dp_{i-j} \]

一晃眼，以为 \(i\) 的转移和所有的 \(j<i\) 都有关。但实际上需要的只是一个前缀和，所以直接考虑在矩阵里多维护一个 \(S_{i-2}\) 即可。

枚举 \(x,y\) 的时候要保证 \((x,y)=1\)，所以有复杂度 \(O(r\log r+\log n\log^3 r)\)

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
using namespace std;

typedef long long ll;

inline ll read(){
    ll x=0,flag=1; char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')flag=0;c=getchar();}
    while(c>='0'&&c<='9'){x=(x<<1)+(x<<3)+c-48;c=getchar();}
    return flag? x:-x;
}

const int N=24;
const int Mod=1e9+7;

struct Matrix{
    ll a[N+1][N+1];
    Matrix(){memset(a,0,sizeof(a));}
    Matrix operator *(const Matrix &X) const{
        Matrix C;
        for(int k=0;k<=N;k++)
            for(int i=0;i<=N;i++)
                for(int j=0;j<=N;j++)
                    C.a[i][j]=(C.a[i][j]+a[i][k]*X.a[k][j]%Mod)%Mod;
        return C;
    }
    void print(){
        for(int i=0;i<=N;i++){
            for(int j=0;j<=N;j++)
                printf("%lld ",a[i][j]);
            putchar('\n');
        }
    }
};

ll n,l,r,cnt[N];

ll gcd(ll x,ll y){return y? gcd(y,x%y):x;}

int main(){
    freopen("excellent.in","r",stdin);
    freopen("excellent.out","w",stdout);
    int T=read();
    while(T--){
        n=read(),l=read(),r=read();
        for(int i=1;i<N;i++) cnt[i]=0;
        int rg=(int)(sqrt(r)+0.5);
        cnt[1]=r-l+1,cnt[2]=(r-l+1)*(r-l)%Mod;
        for(int i=1;i<=rg;i++)
            for(int j=1;j<i;j++){
                if(gcd(i,j)!=1) continue;
                for(int pw=3,x=i*i,y=j*j;x<=r;x*=i,y*=j,pw++)
                    cnt[pw]=(cnt[pw]+max(0ll,r/x-(l-1)/y))%Mod;
            }
        for(int i=3;i<N;i++) cnt[i]=(cnt[i]<<1)%Mod;
        Matrix A,ret;
        for(int i=1;i<=N;i++) A.a[i-1][0]=cnt[i];
        for(int i=1;i<N;i++) A.a[i-1][i]=1;
        A.a[N][N]=A.a[0][N]=1,A.a[N][0]=r-l+1;
    //    A.print();
        for(int i=0;i<=N;i++) ret.a[i][i]=1;
        while(n){ if(n&1) ret=A*ret; A=A*A,n>>=1; }
        printf("%lld\n",ret.a[0][0]);
    }
}

标签：lfloor,log,leq,公比,拆分,序列,include,NYG,dp
来源： https://www.cnblogs.com/wwlwQWQ/p/15286674.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

NYG的序列拆分

Description

Solution