ABC129F题解

ABC129F Takahashi’s Basics in Education and Learning

题目大意

有一个等差数列，有$L$个元素：$s_0,s_1,s_2,\dots ,s_{L-1}$.初始元素为$A$，公差为$B$，$s_i=A+B\times i$.将这些元素从左到右拼接为一个整数，如$3,7,11,15,19$拼接起来得到$37111519$这个整数。输出拼接后的整数模$M$的结果，数据保证等差数列中的前$L$项都不超过$10^{18}$.

题解

当存在一对整数$(l,r)$，满足$\forall i\in [l,r],10^{d-1}\le s_i<10^d$，设拼接到第$i-1$个整数时这个数为$x$，则拼接到第$i$个整数时这个数为$x+s_i$。我们构造一个向量$(x,s_i)$和一个矩阵$C$，使得$(x,s_i)\times C=(x\times 10^d,s_i+B)$，其中$s_i+B$就是$s_{i+1}$。

但在构造矩阵的时候，我们发现难以将$s_i$变为$s_i+B$，所以我们将向量变为$(x,s_i,B)$，那么：

$$(x,s_i,B)\times \left[\begin{array}{ccc}10^d& 0& 0\\ 1& 1& 0\\ 0& 1& 1\end{array}\right]=(x\times 10^d+s_i,s_i+B,B)$$

那么，用矩阵快速幂就能求出对于每个$d$的答案了。

那么，怎么求对于每个$d$，矩阵要乘的次数呢？

设当前这一段每个数都有$d$位的范围为$(l_d,r_d)$，对于每个$d$，显然$i=l_d$是满足$A+B\times i\ge 10^d$的最小值。变形之后就变为$i\ge \frac{10^d-A}{B}$。那么$l_d=\lceil \frac{10^d-A}{B}\rceil$。当然，有时候$\lceil \frac{10^d-A}{B}\rceil$为负，所以得出的$l_d$要与$0$取$\max$。显然，此时的$r_d=l_{d+1}-1$。

时间复杂度为$O(18\times 27\log L)$。

code

#include<bits/stdc++.h>
using namespace std;
long long l,a,b,p,ans,mi[25],h[25];
struct matrix{
	long long a[4][4];
	matrix operator*(const matrix ax)const{
		matrix cx;
		memset(cx.a,0,sizeof(cx.a));
		for(int i=1;i<=3;i++){
			for(int j=1;j<=3;j++){
				for(int k=1;k<=3;k++){
					cx.a[i][j]=(cx.a[i][j]%p+a[i][k]%p*(ax.a[k][j]%p)%p)%p;
				}
			}
		}
		return cx;
	}
}re,w;
matrix ksm(matrix t,long long v){
	matrix c;
	memset(c.a,0,sizeof(c.a));
	if(v==0){
		c.a[1][1]=c.a[2][2]=c.a[3][3]=1;
		return c;
	}
	c=ksm(t,v/2);
	c=c*c;
	if(v%2) c=c*t;
	return c;
}
int main()
{
	scanf("%lld%lld%lld%lld",&l,&a,&b,&p);
	memset(re.a,0,sizeof(re.a));
	memset(w.a,0,sizeof(w.a));
	mi[0]=1;
	for(int i=1;i<=18;i++) mi[i]=mi[i-1]*10;
	for(int i=1;i<=18;i++){
		h[i]=(mi[i]-a)/b;
		if((mi[i]-a)%b>0) ++h[i];
		h[i]=max(h[i],0ll); 
	}
	w.a[2][1]=w.a[2][2]=w.a[3][3]=1;
	w.a[3][2]=1;
	re.a[1][2]=a;re.a[1][3]=b;
	for(int i=0;i<=17;i++){
		w.a[1][1]=mi[i+1];
		re=re*ksm(w,min(h[i+1]-h[i],l));
		l-=h[i+1]-h[i];
		if(l<=0) break;
	}
	printf("%lld",re.a[1][1]);
	return 0;
}