2021牛客多校3 E

题目大意：

有$t(\le 1e5)$次询问，每次询问给定$n(\le 1e18)$，给出有多少组$(x,y)$满足$(xy+1)|(x^2+y^2)$

解题思路：

• 设$(x^2+y^2)=k(xy+1)$
• 固定x不懂，由韦达定理，存在另一个$y^{\prime }$满足$y+y^{\prime }=kx，y{y}^{\prime }=x^2-k$ 不妨设$x\le y$
• $(x,y^{\prime })$也是一组解，并且$y^{\prime }\ge 0,y^{\prime }\le y$
• 因为方程$x$和$y$是对称的，所以可以设小的$y^{\prime }$为对称轴，得到更小的$x^{\prime }$和其对应，大概就类似如下图递降下去直到$y^{\prime }=0$
  
• 最后要求$y=kx,k=x^2$
• 然后根据这个解再逆推出所有解
• $(x,y^{\prime })\to (kx-y^{\prime },x)\to ...$

AC代码：

#include <bits/stdc++.h>
#define ft first
#define sd second
#define pb push_back
#define IOS ios::sync_with_stdio(false), cin.tie(0), cout.tie(0) //不能跟puts混用
#define seteps(N) fixed << setprecision(N)
#define endl "\n"
const int maxn = 1e6 + 10;
using namespace std;
typedef long long ll;
typedef double db;
typedef pair<int, int> pii;
const ll mod = 1e9 + 7;
const ll M = 1e18;
vector <ll> ans;
// __int128 x;
int main() {
	for (ll k = 2; k * k * k <= M; k++) { //k从2开始，所以最后答案要考虑+1，因为k=1没有考虑
		ll now = k * k * k, pre = k;
		ll t;
		__int128 tmp = k * k;
		while (1) {
			ans.pb(now);
			if (tmp * now - pre > M) break;
			t = now;
			now = tmp * now - pre;
			pre = t;
		}
	}
	sort (ans.begin(), ans.end());
	int t;
	ll n;
	cin >> t;
	while (t--) {
		cin >> n;
		cout << upper_bound(ans.begin(), ans.end(), n) - ans.begin() + 1 << endl;
	}
	
}