【bzoj】2326 [HNOI2011]数学作业

时间:2021-07-04 09:01:16

【题意】给定n和m,求1~n从高位到低位连接%m的结果。n=11时,ans=1234567891011%m。n<=10^18,m<=10^9。

【算法】递推+矩阵快速幂

【题解】

考虑枚举位数个数k,对于不同的k单独递推,设f[i]表示1~i的答案,则有:

$$f_n=f_{n-1}*10^k+i$$

转化为矩阵递推式,则有:

$$\begin{vmatrix}10^k & 1 & 1\\ 0 & 1 & 1\\0 & 0 & 1\end{vmatrix} \times \begin{vmatrix}f_n \\ n\\1 \end{vmatrix}=\begin{vmatrix}f_{n+1}\\n+1\\1\end{vmatrix}$$

转化为幂形式,则有:

$$\begin{vmatrix}10^k & 1 & 1\\ 0 & 1 & 1\\0 & 0 & 1\end{vmatrix}^n \times \begin{vmatrix}f_i \\ i\\1 \end{vmatrix}=\begin{vmatrix}f_{i+n}\\i+n\\1\end{vmatrix}$$

分段进行矩阵快速幂即可。

注意读入的n是long long。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=;
int c[N][N],ANS[N][N],A[N][N],m;
ll n;///
void multply(int a[N][N],int b[N][N]){
for(int i=;i<=;i++){
for(int j=;j<=;j++){
c[i][j]=;
for(int k=;k<=;k++){
c[i][j]=(c[i][j]+1ll*a[i][k]*b[k][j]%m)%m;
}
}
}
for(int i=;i<=;i++)for(int j=;j<=;j++)b[i][j]=c[i][j];
}
void solve(ll p,int &f,int x,int k){
memset(A,,sizeof(A));
A[][]=A[][]=A[][]=A[][]=A[][]=;A[][]=k;
ANS[][]=f;ANS[][]=x%m;ANS[][]=;
while(p){
if(p&)multply(A,ANS);
multply(A,A);
p>>=;
}
f=ANS[][];
}
int main(){
scanf("%lld%d",&n,&m);
ll x=,y=;
int k=%m,ans=;
while(x+y<n){
solve(y,ans,x%m,k);//
x+=y;y*=;k=1ll*k*%m;
}
solve(n-x,ans,x%m,k);
printf("%d",ans%m);
return ;
}