这个代码的效率较低，如何优化以提高性能？

Question

这个代码的效率较低，如何优化以提高性能？

def calculate_sum(n):
    total = 0
    for i in range(n+1):
        total += i
    return total

result = calculate_sum(1000000)
print(result)

Answer 1

让阿豪来帮你解答，本回答参考chatgpt3.5编写提供，如果还有疑问可以评论或留言

首先，这段代码的效率较低主要是因为使用了循环来逐个相加，这样的算法的时间复杂度为O(n)，可以通过数学公式来优化以提高性能。 一种解决方法是通过数学公式直接计算1到n的和，而不是使用循环逐个相加。对于1到n的和，数学公式为：sum = n * (n + 1) / 2。 下面是优化后的代码示例：

def calculate_sum(n):
    total = n * (n + 1) / 2
    return total
result = calculate_sum(1000000)
print(result)

这样可以大大提高代码的性能，因为数学公式的时间复杂度为O(1)。