python为什么会超时???
查看原帖
python为什么会超时???
420021
kfjla89楼主2022/4/7 15:59

全部样例超时,嗯。。。。主要代码如下:

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])

2022/4/7 15:59
加载中...