Sol
数学.
跟圆上的整点一样...TA写了个积性函数的算法...以后再说吧...
\(x^2+3y^2=r^2\)
\(3y^2=r^2-x^2\)
\(3y^2=(r-x)(r+x)\)
\(y^2=\frac{1}{3}(r-x)(r+x)\)
\(d=(r-x)(r+x)\)
\(r-x=3du^2,r+x=dv^2\) 这里 \(r-x\) 和 \(r+x\) 并没有什么区别.
\(2r=d(3u^2+v^2)\)
枚举 \(d\) 和 \(u\)
感觉复杂度是\(O(n^{\frac{3}{4}})\)
但是可以跑最大数据的说.
Code
/**************************************************************
Problem: 4544
User: BeiYu
Language: C++
Result: Accepted
Time:8568 ms
Memory:1300 kb
****************************************************************/
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<vector>
#include<utility>
#include<iostream>
using namespace std;
typedef long long LL;
#define debug(a) cout<<#a<<"="<<a<<" "
#define mpr(a,b) make_pair(a,b)
LL T,r,n,ans;
inline LL in(LL x=0,char ch=getchar()){ while(ch>'9'||ch<'0') ch=getchar();
while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();return x; }
vector<pair<LL,LL> > p;
LL calc(LL d){
LL res=0,m=n/d;
// cout<<"*************"<<endl;
// debug(m),debug(d);cout<<endl;
for(LL u=1,v;u*u*3<=m;u++){
v=sqrt(m-3*u*u+0.5);
// debug(u),debug(v),debug(3*v*v+u*u),cout<<endl;
// if(u>v) break;
if(v*v+u*u*3==m&&__gcd(v*v,u*u*3)==1) res++;
// cout<<"get!",debug(d*u*u*3),debug(d*v*v),debug(d*u*u*3+d*v*v)<<endl;
// p.push_back(mpr(d*u*u*3,d*v*v));
}return res;
}
int main(){
// freopen("in.in","r",stdin);
for(T=in();T--;){
r=in(),n=r<<1,ans=0;
for(LL d=1;d*d<=n;d++) if(n%d==0){
if(d*d==n) ans+=calc(d);
else ans+=calc(d)+calc(n/d);
}
cout<<ans*4+2<<endl;
// sort(p.begin(),p.end());
// for(int i=0;i<p.size();i++) cout<<p[i].first<<" "<<p[i].second<<endl;
}
return 0;
}