UVa 106 - Fermat vs. Pythagoras

/*
数学：毕达哥拉斯数, GCD
如果u, v不含公因子，且都不是奇数，则 u*u - v*v, 2uv, u*u + v*v 是不含公因子的毕达哥拉斯数。参考《什么是数学》
*/
#include <cstdio>
#include <cmath>
#include <cstring>
using namespace std;
const int MAXN = 1000004;
int vis[MAXN];
int n;

int gcd(int a, int b)
{
    return b==0 ? a : gcd(b, a%b);
}

void set_vis(int a, int b, int c)
{
    for(int k=1; c*k<=n; k++) {
        vis[a*k] = vis[b*k] = vis[c*k] = 1;
    }
}

int count_vis()
{
    int cnt = 0;
    for(int i=1; i<=n; i++) {
        if(!vis[i]) {
            cnt++;
        } else {
            vis[i] = 0;
        }
    }
    return cnt;
}

int main() {
    #ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    #endif
    while(scanf("%d", &n) == 1) {
        int m = sqrt(n+1);
        int cnt = 0;
        for(int i=1; i<m; i++) {
            for(int j=i+1; j<=m; j+=2) {
                if(i*i + j*j > n) break;
                if(i!=1 && gcd(j, i)!=1) continue;
                cnt++;
                set_vis(j*j - i*i, 2*i*j, i*i + j*j);
            }
        }
        printf("%d %d\n", cnt, count_vis());
    }
    return 0;
}