BZOJ 1041 [HAOI2008]圆上的整点

1041: [HAOI2008]圆上的整点

Description

　　求一个给定的圆(x^2+y^2=r^2)，在圆周上有多少个点的坐标是整数。

Input

　　只有一个正整数n,n<=2000 000 000

Output

　　整点个数

Sample Input

4

Sample Output

4

---

　　鸣谢：http://blog.csdn.net/csyzcyj/article/details/10044629  http://hzwer.com/1457.html

　　这么一到水题竟然卡了我一晚上，想起来确实是不可思议。

　　样例图示：

　　我们可以十分清晰的发现，这似乎就是一道算几题，但我们又很难下手。如果暴力枚举每个整数x，很明显会TLE。那怎么办呢？

　　

　　右上图可知，我们可以枚举2*d的因数，这是O（sqrt(n)），再去枚举界的个数。因为此处前提为x>0,y>0,所以最后结果为ans*4+4。

　　因为涉及实数，所以屡调不顺，下次需多加注意。（floor一个实数结果仍为实数)

　　

/************************************************************** 
    Problem: 1041 
    User: Doggu 
    Language: C++ 
    Result: Accepted 
    Time:104 ms 
    Memory:828 kb 
****************************************************************/
  
#include <cstdio>  
#include <cmath>  
long long n, ans;  
inline long long gcd(long long a,long long b) {return b==0?a:gcd(b,a%b);}  
int main() {  
    scanf("%lld",&n);  
    long long cou, b;  
    double t;  
    for( long long i = 1; i*i <= 2*n; i++ ) {  
        if(2*n%i==0) {  
            cou=i;  
            for( long long a = 1; 2*a*a < cou; a++ ) {  
                t=sqrt(cou-a*a);  
                if(t==floor(t)) {  
                    b=(long long)floor(t);  
                    if(a!=b&&gcd(a,b)==1) ans++;   
                }  
            }  
            cou=2*n/i;  
            if(2*n/i==i) continue;  
            for( long long a = 1; 2*a*a < cou; a++ ) {  
                t=sqrt(cou-a*a);  
                if(t==floor(t)) {  
                    b=(long long)floor(t);  
                    if(a!=b&&gcd(a,b)==1) ans++;   
                }  
            }  
        }  
    }  
    printf("%lld\n",ans*4+4);  
    return 0;  
}