全部样例超时,嗯。。。。主要代码如下:
def dfs():
global dp
n=32768
dp[0][0]=1
for i in range(1,int(sqrt(n))+1):
for j in range(1,5):
for k in range(i**2,n+1):
dp[j][k]=dp[j][k]+dp[j-1][k-i**2]
dp=[defaultdict(int) for i in range(5)]
dfs()
t=int(input())
for k in range(t):
r=int(input())
print(dp[1][r]+dp[2][r]+dp[3][r]+dp[4][r])