洛谷P1586 四方定理

P1586 四方定理

题目描述

#include <bits/stdc++.h>

inline void read(long long &x)
{
x = 0;char ch = getchar();char c = ch;
while(ch > '9' || ch < '0')c = ch, ch = getchar();
while(ch <= '9' && ch >= '0')x = x * 10 + ch - '0', ch = getchar();
if(c == '-')x = -x;
}
const int INF = 0x3f3f3f3f;
const int MAXN = 1 << 15;

long long t,n,dp[MAXN + 10][5];
long long ans;

int main()
{
read(t);
for(;t;--t)
{
ans = 0;
read(n);
for(int i = 0;i * i <= n;i ++)
{
int tmp1 = i * i;
for(int j = i;tmp1 + j * j <= n;j ++)
{
int tmp2 = tmp1 + j * j;
for(int k = j;tmp2 + k * k <= n;k ++)
{
int tmp3 = tmp2 + k * k;
int c = n - tmp3;
if(c < k * k)break;
int m = sqrt(c);
if(m * m == c)ans ++;
}
}
}
printf("%lld\n", ans);
}
return 0;
}