题目描述
输入描述
输出描述
输入样例
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;
}