2021牛客多校#3 Math

原题链接

		https://ac.nowcoder.com/acm/contest/11254/E

题目大意

给定$n（1\le n\le 10^{18}）$，求满足$xy+1|x^2+y^2$的正整数对$（x，y）（1\le x\le y\le n）$的数量。

题解

设$k(xy+1)=x^2+y^2$。
根据韦达定理可得$（x，kx-y）$是一组解，且$k$为平方数。

不难发现当$y=x^3$时，$（x，y）$满足条件且此时$k=x^2$。

于是我们只要枚举第一组解$（x，x^3）$并根据上述关系式进一步计算更多的解即可。

参考代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll MAXN=1e18;
int T;
ll n,m=0,v[5000010];
int main()
{
	v[m++]=1;
	for(ll i=2;i*i*i<=MAXN;i++)
	{
		ll x=i,y=i*i*i;
		v[m++]=y;
		while(y<=(MAXN+x)/i/i)
		{
			x=i*i*y-x;
			swap(x,y);
			v[m++]=y;
		}
	}
	sort(v,v+m);//从小到大进行排序方便二分找答案
	scanf("%d",&T);
	while(T--)
	{
		scanf("%lld",&n);
		printf("%lld\n",upper_bound(v,v+m,n)-v);
		//二分找到的下标即是需要输出的正整数对的数量
	}
}