math（2021牛客多校第三场E）

题目描述

输入描述

输出描述

输入样例

10

10

100

1000

10000

100000

114514

1919810

20190104

123123123123

10000001000000

输出样例

2
5
14
31
65
67
158
326
5226
22091

第一眼就被表达式难倒的题，正确解读是 (x^2 + y^2) mod (xy+1)=0,数据范围1e18可直接排除暴力做法，但可以先通过打表 n = 10000 时满足条件的数对寻找规律。

//打表代码
for(int i=1;i<=10000;i++)
	for(int j=i;j<=10000;j++)
		if((i*i+j*j)%(i*j+1)==0)
			printf("%d %d\n",i,j);

仔细观察后可以看出规律：
(2,8)->(8,30)->(30,112)->(112,418)->(418,1560)->(1560,5822)
(3,27)->(27,240)->(240,2133)
(4,64)->(64,1020)
(5,125)->(125,3120)
(6,216)->(216,7770)

该情景类似于Hash表处理冲突时的拉链法，由2开始的自然数都可引出满足要求的数对，且前一数对的第二个数恰好也是后一数对的第一个数，直到下一数对超过n为止；

如果以数对形式观察难以找出他们之间的规律的话，可以将数字单独提出：
2->8->30->112->418->1560->5822;
3->27->240->2133;
4->64->1020;
5->125->3120;
6->216->7770;

不难发现 a [ n ] = i * i * a [ n - 1 ] - a [ n - 2 ] ;

其中 a [ 0 ] = 0 , a [ 1 ] = i (i为每行第一位数：2,3,4,5,6……) ，而 a [ 2 ] 恰好为 a [ 1 ] 的立方，而由于题目数据范围为1e18，因此 i 的范围为1e6。

解题思路即为预处理并将所有符合要求的数存入数组中，通过内置二分计算满足条件n的数字个数即可。

特别需要注意的是，本题ull也有爆的可能（ull的范围为1e19），而_int128的范围为4e22，可有效避免爆数据。

实现代码

#include <bits/stdc++.h>
#define ll long long
typedef unsigned long long ull;
using namespace std;
const int N=1e7+10;
const ll mod=998244353;
__int128 a[N];
int main()
{
	int t;
    scanf("%d",&t);
    __int128 c,rear,num=0;
    a[num++]=1;
    for(ll i=2;i<=1000000;i++){
        c=i*i*i,rear=i;
        while(c<=1e18){
            a[num++]=c;
            __int128 cmp=c;
            c=i*i*c-rear;
            rear=cmp;
        }
    }
    sort(a,a+num);

	while(t--){
		ll n;
        cin>>n;
        int ans=lower_bound(a, a+num,n)-a;
        if(a[ans]==n)ans++;
        cout<<ans<<endl;
	}
    return 0;
}