牛客2021暑期训练3-E-Math

牛客2021暑期训练3-E-Math

题目链接

题意

给定 n(n≤1e18)，求 (x,y) 数量，其中 xy+1 | x²+y² (1≤x≤y≤n) (x²+y²能被xy+1整除)

题解

先看个公式证明

回到本题，假设我们已知数对 (x,y) 为一组解，必然还有一组解为 (x,y’) ，又由上面的证明可知 y’<x≤y，而题目要求 x≤y，因此 (x,y’)->(y’,x)，那么此时两组符合要求的解则为 (y’,x) (x,y) ，同时 y=k*x-y’

通过上面的公式可以求出一组解 (x,x³) k=x²，那么我们将 (x,x³) 视为 (y’,x)，我们就可以顺势向上求出 (x,y)

怎么求呢。我们已知 (x,y=x³) 为一组解，那么 (y=x³,k*x-y’=x⁵-x) 则为另一组解，不断地线上推，只要记录满足的 y 就可以了

小提醒，用__int128

代码

#include<bits/stdc++.h>
#define LL __int128
using namespace std;
const LL N=1e6,M=1e18;
int T;
LL n;
vector<LL>v;
inline LL read()
{
	LL w=1,s=0;
	char ch=getchar();
	while(ch<'0'||ch>'9') {if(ch=='-') w=-1; ch=getchar();}
	while(ch>='0'&&ch<='9') {s=s*10+ch-'0'; ch=getchar();}
	return w*s;
}
int main()
{
	v.push_back(1LL);
	for(LL i=2;i<=N;i++)
	{
		LL x=i,y=i*i*i,k=i*i;
		while(y<=M)
		{
			v.push_back(y);
			LL temp=x;
			x=y;
			y=k*x-temp;
		}
	}
	sort(v.begin(),v.end());
	cin>>T;
	while(T--)
	{
		n=read();
		int p=upper_bound(v.begin(),v.end(),n)-v.begin();
		cout<<p<<endl;
	}
	return 0;
}