def main0(n,d): ans=0 for x in range(1,n+1): for y in range(1,n+1): for z in range(1,n+1): for w in range(1,n+1): if x**2+y**2+z**2==d+w**2: ans+=1 return ans # 1<=N<=2000 # 0<=D<=10**6 def main1(n,d): mat0=[0]*(1+2*n**2) for x in range(1,n+1): for y in range(1,n+1): mat0[x**2+y**2]+=1 mat1=[0]*(1+2*n**2) for x in range(1,n+1): for y in range(1,n+1): if 0