思想
使用动规,具体步骤如下:
1、定义状态:我们可以定义一个一维数组 dp,其中 dp[i] 表示以第 i 个元素结尾的连续子数组的最大和。
2、状态转移方程:考虑 dp[i],它可以由以下两种情况中的较大值得到:
① 第 i 个元素自成一个子数组,即 nums[i];
②第 i 个元素和前面的某个元素结合,形成一个新的子数组。如果 dp[i-1] 大于 0,则将 nums[i] 加上 dp[i-1]。
所以,状态转移方程为:dp[i] = max(nums[i], nums[i] + dp[i-1])。
3、初始状态:dp[0] = nums[0],即以第一个元素结尾的连续子数组的最大和就是第一个元素本身。
4、最终结果:最大和即为 dp 数组中的最大值。
题目描述
给你一个整数数组 nums ,请你找出一个具有最大和的连续子数组(子数组最少包含一个元素),返回其最大和。
子数组是数组中的一个连续部分。
示例 1:
输入:nums = [-2,1,-3,4,-1,2,1,-5,4]
输出:6
解释:连续子数组 [4,-1,2,1] 的和最大,为 6 。
示例 2:
输入:nums = [1]
输出:1
示例 3:
输入:nums = [5,4,-1,7,8]
输出:23
完整代码
// 动规
class Solution {
public:
int maxSubArray(vector<int>& nums) {
if(nums.size() == 0) return 0;
vector<int> dp(nums.size());
dp[0] = nums[0];
for(int i = 1; i < nums.size(); i++)
dp[i] = max(nums[i], dp[i - 1] + nums[i]);
return *max_element(dp.begin(), dp.end());
}
};
// 贪心
class Solution {
public:
int maxSubArray(vector<int>& nums) {
int result = INT32_MIN;
int count = 0;
for(int i = 0; i < nums.size(); i++){
count += nums[i];
if(count > result) result = count;
if(count <= 0) count = 0;
}
return result;
}
};