小编Pra*_*lar的帖子

在这段代码中，我正在计算n数字的平方和n数字总和的平方之间的差。

例子 ： n=3, (1+2+3)^2 -(1^2+2^2+3^2) =22

def sum_square_diff(num):
    sum1=0
    sum2=0
    for i in range(1,num+1):
        sum1 +=i**2
        sum2 +=i
    sum2=sum2**2
    diff=sum2-sum1
    return diff

if __name__=="__main__":
    n=int(input())
    for i in range(n):
        num=int(input())
        result=sum_square_diff(num)
        print(result)

这段代码是正确的，但完成执行需要太多时间。