把博客园图标替换成自己的图标
把博客园图标替换成自己的图标end

【洛谷7325】[WC2021] 斐波那契(数论)

点此看题面

  • 定义\(F_0=a,F_1=b,F_i=(F_{i-1}+F_{i-2})\ mod\ m\)
  • 初始给定\(m\),每次询问给出\(a,b\),求最小的\(n\)满足\(F_n=0\)
  • \(n,m\le10^5\)

斐波那契数

首先我们发现\(F_i=aFib_{i-2}+bFib_{i-1}\)(可以归纳证明)。

那么现在就是要找到最小的\(n\)使得\(aFib_{n-2}+bFib_{n-1}\equiv0(mod\ m)\)

方便起见令把原本的\(b\)变成\(-b\),那么就得到\(aFib_{n-2}\equiv bFib_{n-1}(mod\ m)\)

如果\(m\)是质数,可以得到\(\frac ab\equiv\frac{Fib_{n-1}}{Fib_{n-2}}(mod\ m)\),直接预处理出\(p_x\)表示\(\frac{Fib_{n-1}}{Fib_{n-2}}\equiv x(mod\ m)\)的最小的\(n\),询问时查询\(p_{\frac ab}\)即可。

可惜\(m\)不一定是质数,但此时我们需要注意一个关键性质:\(Fib_{n-1}\perp Fib_{n-2}\)

证明就是模拟辗转相减法的过程,\(gcd(Fib_{n-1},Fib_{n-2})=gcd(Fib_{n-2},Fib_{n-3})=...=gcd(1,0)=1\)

因此,我们首先约去\(gcd(a,b,m)\),得到\(a'Fib_{n-2}\equiv b'Fib_{n-1}(mod\ m')\)

由于模\(m\)意义下两式相等,必须满足\(gcd(a'Fib_{n-2},m')=gcd(b'Fib_{n-1},m')\)

\(p=gcd(a',m'),q=gcd(b',m')\),因为已经约去过\(gcd(a,b,m)\),显然有\(p\perp q\)

因此,我们得到\(gcd(Fib_{n-1},m')=p,gcd(Fib_{n-2},m')=q\)

在两式中同时约去\(pq\)得到\(a''Fib_{n-2}'\equiv b''Fib_{n-1}'(mod\ {\frac{m'}{pq}})\)

此时由于所有数都与模数互质,可以套用先前质数的做法,得到\(\frac{a''}{b''}\equiv\frac{Fib_{n-1}'}{Fib_{n-2}'}(mod\ \frac{m'}{pq})\)

所以只要对于每个\(m'\)的每个三元组\((x,y,v)\)记录满足\(gcd(Fib_{n-2},m')=x\)\(gcd(Fib_{n-1},m')=y\)\(\frac{Fib_{n-1}\div y}{Fib_{n-2}\div x}\equiv v(mod\ \frac{m'}{xy})\)的最小的\(n\)即可。

由于斐波那契数在模\(m\)意义下的循环节是\(O(m)\)的,直接暴力枚举直至产生循环为止即可。

代码:\(O(\sigma(m)logm)\)

#include<bits/stdc++.h>
#define Tp template<typename Ty>
#define Ts template<typename Ty,typename... Ar>
#define Reg register
#define RI Reg int
#define Con const
#define CI Con int&
#define I inline
#define W while
#define N 100000
using namespace std;
int m;I int gcd(CI x,CI y) {return y?gcd(y,x%y):x;}
I void exgcd(CI x,CI y,int& a,int& b) {y?(exgcd(y,x%y,b,a),b-=x/y*a):(a=1,b=0);}
I int Inv(CI x,CI y) {RI a,b;return exgcd(x,y,a,b),(a%y+y)%y;}
struct Data {int gx,gy,v;I Data(CI x=0,CI y=0,CI z=0):gx(x),gy(y),v(z){}
	I bool operator < (Con Data& o) Con {return gx^o.gx?gx<o.gx:(gy^o.gy?gy<o.gy:v<o.v);}};map<Data,int> p[N+5];
namespace FastIO
{
	#define FS 100000
	#define tc() (FA==FB&&(FB=(FA=FI)+fread(FI,1,FS,stdin),FA==FB)?EOF:*FA++)
	#define pc(c) (FC==FE&&(clear(),0),*FC++=c)
	int OT;char oc,FI[FS],FO[FS],OS[FS],*FA=FI,*FB=FI,*FC=FO,*FE=FO+FS;
	I void clear() {fwrite(FO,1,FC-FO,stdout),FC=FO;}
	Tp I void read(Ty& x) {x=0;W(!isdigit(oc=tc()));W(x=(x<<3)+(x<<1)+(oc&15),isdigit(oc=tc()));}
	Ts I void read(Ty& x,Ar&... y) {read(x),read(y...);}
	Tp I void writeln(Ty x) {W(OS[++OT]=x%10+48,x/=10);W(OT) pc(OS[OT--]);pc('\n');}
	I void NA() {pc('-'),pc('1'),pc('\n');}
}using namespace FastIO;
int main()
{
	RI Qt,i,j,x,y,z,gx,gy,v,m_;for(read(Qt,m),i=2;i<=m;++i) if(!(m%i)) for(j=2,x=y=1;;++j)//对于m的每个因数暴力预处理
	{
		m_=i/(gx=gcd(x,i))/(gy=gcd(y,i)),v=1LL*(y/gy)*Inv(x/gx,m_)%m_,!p[i].count(Data(gx,gy,v))&&(p[i][Data(gx,gy,v)]=j);//记录每个三元组最早的n
		if(z=(x+y)%i,x=y,y=z,x==1&&y==1) break;//如果产生循环节
	}
	RI g;W(Qt--)
	{
		if(read(x,y),!x) {writeln(0);continue;}if(!y) {writeln(1);continue;}y=(m-y)%m;//特判x=0或y=0,将y变成-y
		g=gcd(gcd(x,y),m),m_=m/g/(gx=gcd(x/=g,m/g))/(gy=gcd(y/=g,m/g)),v=1LL*(x/gx)*Inv(y/gy,m_)%m_;//计算出三元组
		p[m/g].count(Data(gy,gx,v))?writeln(p[m/g][Data(gy,gx,v)]):NA();//如果存在就输出,否则无解
	}return clear(),0;
}
posted @ 2021-05-17 08:34  TheLostWeak  阅读(68)  评论(0编辑  收藏  举报