下面我将分别使用动态规划和前缀和来分别做这道题,先来看题目:
给你一个整数数组 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
提示:
1 <= nums.length <= 105
-104 <= nums[i] <= 104
1、使用动态规划:
int maxSubArray(vector<int>& nums) {
// 动态规划
int maxi = INT_MIN;
if (nums.size() == 1) {
return nums[0];
}
vector<int> dp(nums.size(), 0);
dp[0]=nums[0];
maxi=nums[0];
for (int i = 1; i < nums.size(); i++) {
dp[i] = max(nums[i], nums[i] + dp[i - 1]);
maxi = max(maxi, dp[i]);
}
return maxi;
}
2、使用前缀和:
int maxSubArray(vector<int>& nums) {
// 前缀和
if (nums.size() == 1) {
return nums[0];
}
int maxi = nums[0], mini = 0, pre = 0;
for (int i = 0; i < nums.size(); i++) {
pre += nums[i];
maxi = max(maxi, pre - mini);
mini = min(mini, pre);
}
return maxi;
}
如果使用两个for循环的话会超时,所以这里用mini来记录前面最小的前缀和。