1、题目描述
输入一个整型数组,数组中的一个或连续多个整数组成一个子数组。求所有子数组的和的最大值。
要求时间复杂度为O(n)。
示例1:
输入: nums = [-2,1,-3,4,-1,2,1,-5,4]
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大,为 6。
2、VS2019上运行
方法一:动态规划
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
class Solution {
public:
int maxSubArray(vector<int>& nums) {
int pre = 0, maxAns = nums[0]; // 初始化pre和maxAns
for (const auto& x : nums) {
pre = max(pre + x, x); // 更新pre,取pre + x和x中的较大值
maxAns = max(maxAns, pre); // 更新maxAns,取maxAns和pre中的较大值
}
return maxAns;
}
};
int main() {
vector<int> nums = { -2,1,-3,4,-1,2,1,-5,4 }; // 示例输入
Solution solution;
int result = solution.maxSubArray(nums);
cout << "Maximum subarray sum: " << result << endl;
return 0;
}
Maximum subarray sum: 6
3、解题思路
- 首先,使用两个变量 pre 和 maxAns 来记录当前位置之前的连续子数组的最大和和整个数组的最大连续子数组的和。初始时,将它们都设置为数组第一个元素 nums[0]。
- 然后,通过遍历数组中的每个元素,不断更新 pre 和 maxAns 的值。
- 对于每个元素 nums[i],更新 pre 的规则是取 pre + nums[i] 和 nums[i] 中的较大值,表示以当前位置元素结尾的连续子数组的最大和。
- 然后,将 maxAns 更新为 max(maxAns, pre),将 pre 和 maxAns 中的较大值赋给 maxAns,表示更新全局最大连续子数组的和。
- 通过对整个数组进行遍历,不断更新 pre 和 maxAns,最终得到整个数组中连续子数组的最大和就保存在 maxAns 变量中。