通过将 Array 元素替换为 Square 可以得到的最大子数组总和

有一个由整数组成的数组 Array，现在可以将其中的任意一个元素替换为该元素的平方，问替换一个元素后能得到的最大子数组总和是多少？

例如，如果数组 Array 是 [-2, 5, -1, 0, 3, 2, -4]，那么将元素 -2 替换为 4 后，得到的最大子数组是 [16, 25, -1, 0, 3, 2, -4]，总和为 41。

以下是一个可以解决这个问题的 Python 代码片段：

def max_sum_after_replace_square(nums):
    n = len(nums)
    # 动态规划数组 dp
    # dp[i] 表示以 nums[i] 结尾的最大子数组总和
    dp = [0] * n
    dp[0] = nums[0]
    ans = nums[0]
    for i in range(1, n):
        # 计算 dp[i]，有两种情况：
        # 1. 把 nums[i] 替换成它的平方
        # 2. 不替换 nums[i]
        dp[i] = max(dp[i-1] + nums[i]*nums[i], nums[i])
        ans = max(ans, dp[i])
    return ans

这个函数的时间复杂度是 O(n)，其中 n 是数组的长度。