2021牛客多校3 E Math 超详细（应该吧）

E Math

链接

题意：给个n，有1≤x≤y≤n，求有多少个<x,y>对，使得 xy+1｜x²+y²。

（1≤t≤10⁵，1≤n≤10¹⁸）

思路：(typora里公式烂掉了，就直接贴图片了)

综上可知，一组解<x,y>可以生成一组较小解<y’,x>同样符合 xy+1｜x²+y²。

由于固定的x>0，因此x²+y²>0，故k(xy+1)>0，可以得出：任意使得 xy+1｜x²+y²成立的y>=0。

由于kx为固定的值，因此可以发现：对于一个固定的x，最大的y可以由y’=0生成，故kx=y，代入x²+y²=k(xy+1)中，可得k=x²。

代回 kx=y 可得最大的 y=x³，因此可知，对于y=1e18，x最小1e6。

我们打表观察下

1 1
2 8
3 27
4 64
5 125
6 216
7 343
8 30
8 512
9 729
10 1000
11 1331
12 1728
13 2197
14 2744
15 3375
16 4096
17 4913
18 5832
19 6859
20 8000
21 9261
22 10648
23 12167
24 13824
25 15625
26 17576
27 240
27 19683
28 21952
29 24389
30 112
30 27000

可以发现每一组<x,x³>都存在，但有些比较小的对，如<8,30>,<30,112>，实际上这两个对中的4、112就是x=30，k=4时的一组解，而<2,8>,<8,30>，也同样如此，因此可以认为是这些合法的对，都是由<0,2>开始，k=4，每轮将前一个x作为y’，求出新一轮的y，具体过程为：<0,2> -> <2,8> -> <8,30> -> <30,112>…

因此我们可以得出 k 只能为完全平方数，并且由sqrt(k)作为第一个对的x。由于枚举平方不易，因此直接令t=sqrt(k)，即枚举起点x，在转移时新的对的 y 即为 t²x-y’。

当 t 枚举至1e6时，此时第一个合法的对即为<1e6,1e18>，因此之后枚举的 t 的第一个合法对的 y 值，一定大于1e18，因此只需枚举至 t=1e6。

实现时只需将每个合法对的 y 值塞入数组中，最终sort，便可通过二分查找位置，得到值域小于 n 的对的个数了。

#include<iostream>
#include<cstdio>
#include<string>
#include<vector>
#include<cmath>
#include<stack>
#include<queue>
#include<map>
#include<unordered_map>
#include<set>
#include<cstring>
#include<algorithm>
#include<climits>
#include<numeric>
#define int long long
#define pii pair<int,int>
#define IOFast() ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
using namespace std;
const int maxn=1e6+10;
int cnt,ans;
int a[3*maxn];
const int inf=1e18;
int n;
void init(){
	a[0]=1;
	for(int t=2;t<=1e6+1e-4;t++)
	{
		int y_=0,y=t,x;
		while(1)
		{
			x=y;
		//	if(x*t*t-y_>inf||x*t*t-y_<0) break;	
			if(x>(inf+y_)/(t*t)) break;	//这种写法更好，x*t*t可能会爆long long，因此上一行需要特判，此处通过做除法避免这种讨论
			y=y*t*t-y_;
			y_=x;
			a[++cnt]=y;
		}
	}
	sort(a,a+cnt+1);
}
void solve(){
	cin>>n;
	cout<<upper_bound(a,a+cnt+1,n)-a<<'\n';
}
signed main(){
	IOFast();
	init();
	int t;
	cin>>t;
	while(t--)
		solve();
}

代码参考逆十字大佬orz