BZOJ5221[Lydsy2017省队十连测] 偏题

原题链接：https://www.lydsy.com/JudgeOnline/problem.php?id=5221

偏题

Description

斐波那契数列是一个经典递推数列，即
$$Fi{b}_n=Fi{b}_{n-1}+Fin+n-2$$

在这个问题中，定义一个新数列，对于$n\ge 2$有
$$F_n=F_{n-1}+F_{n-2}+2\sqrt{3+F_{n-1}{F}_{n-2}}$$

对于给出的四个整数$F_0,F_1,M,n$，求$F_{n}modM$。

数据保证$\sqrt{3+F_{n-1}{F}_{n-2}}$总是整数。

Input

一行四个整数$F_0,F_1,M,n$。

Output

一行一个整数，表示$F_n$对$M$取模后的值。

Sample Input

1 1 10 5

Sample Output

4

HINT

$F_0=1$
$F_1=1$
$F_2=1+1+2\sqrt{3+1\times 1}=6$
$F_3=6+1+2\sqrt{3+6\times 1}=13$
$F_4=13+6+2\sqrt{3+13\times 6}=37$
$F_5=37+13+2\sqrt{3+37\times 13}=94$

【数据规模】

对于$30\%$的数据，$n\le 20$。

对于$60\%$的数据，$F_0=F_1=1$。

对于$80\%$的数据，$n\le 10^5$。

对于$100\%$的数据，$0\le n\le 10^9，1\le M\le 10^9，1\le F_0\le F_1\le 10^6$。

题解

乍一看可能需要一个二次剩余，然而我们可以先推个式子：

$(\sqrt{3+F_n{F}_{n-1}}{)}^2$
$=3+F_n{F}_{n-1}$
$=3+(F_{n-1}+F_{n-2}+2\sqrt{3+F_{n-1}{F}_{n-2}})F_{n-1}$
$=3+F_{n-1}{F}_{n-2}+2F_{n-1}\sqrt{3+F_{n-1}{F}_{n-2}}+F_{n-1}^2$
$=(\sqrt{3+F_{n-1}{F}_{n-2}}+F_{n-1}{)}^2$

话说为什么$\mathcal{C}\mathcal{S}\mathcal{D}\mathcal{N}$改版以后左对齐的语法不管用了。。。

源码如下：

\begin{align*}
&(\sqrt{3+F_nF_{n-1}})^2\\
&=3+F_nF_{n-1}\\
&=3+(F_{n-1}+F_{n-2}+2\sqrt{3+F_{n-1}F_{n-2}})F_{n-1}\\
&=3+F_{n-1}F_{n-2}+2F_{n-1}\sqrt{3+F_{n-1}F_{n-2}}+F_{n-1}^2\\
&=(\sqrt{3+F_{n-1}F_{n-2}}+F_{n-1})^2
\end{align*}

所以就有$\sqrt{3+F_n{F}_{n-1}}=\sqrt{3+F_{n-1}{F}_{n-2}}+F_{n-1}$，这样我们就可以将$\sqrt{3+F_n{F}_{n-1}}$当做一项放进矩阵里递推了。

初始矩阵：

1 1 1
1 0 0
2 0 1

快速幂出来了以后再乘上一个向量$F_1,F_0,\sqrt{3+F_0{F}_1}$就行了。

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int M=105;
ll n=3,mod,a,f0,f1;
struct matrix{int sq[M][M];matrix(){memset(sq,0,sizeof(sq));}}mat,one;
matrix operator *(matrix a,matrix b)
{matrix c;for(int i=1,j,k;i<=n;++i)for(j=1;j<=n;++j)for(k=1;k<=n;++k)(c.sq[i][j]+=1ll*a.sq[i][k]*b.sq[k][j]%mod)%=mod;return c;}
matrix power(matrix x,ll p){matrix ans=one;for(;p;p>>=1,x=x*x)if(p&1)ans=ans*x;return ans;}
void in(){scanf("%lld%lld%lld%lld",&f0,&f1,&mod,&a);}
void ac()
{
	one.sq[1][1]=one.sq[2][2]=one.sq[3][3]=1;
	mat.sq[1][1]=mat.sq[1][2]=mat.sq[1][3]=mat.sq[2][1]=mat.sq[3][3]=1,mat.sq[3][1]=2;
	mat=power(mat,a-1);
	printf("%lld\n",(mat.sq[1][1]*f1%mod+mat.sq[2][1]*f0%mod+mat.sq[3][1]*(ll)sqrt(3+f0*f1)%mod)%mod);
}
int main(){in();ac();}