一、题目
给定一个整数数组 nums ,找到一个具有最大和的连续子数组(子数组最少包含一个元素),返回其最大和。
示例:
输入: [-2,1,-3,4,-1,2,1,-5,4],
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大,为 6。
进阶:
如果你已经实现复杂度为 O(n) 的解法,尝试使用更为精妙的分治法求解。
二、思路——动态规划
问题的核心在于找到和最大的连续子数组,子数组可以看成由起点和终点确定的一个区间,我们可以先确定其中一个点,然后去找另一个点。比如,我们先确定子数组的终点在 i 位置,然后去找起点的位置。这时候我们要求解的问题就是:在以 i 结尾的所有子数组中,和最大为多少?所以,我们定义状态 d p [ i ] dp[i] dp[i] 表示:“以 i 结尾的所有子数组的最大和”。
求解 d p [ i ] dp[i] dp[i] 可分两种情况:
- 如果以 i - 1 结尾的所有子数组的最大和小于 0,那么以 i 结尾的所有子数组的最大和就是 nums[i] (只包含 nums[i] 这一个元素);
- 如果以 i - 1 结尾的所有子数组的最大和大于 0,那么以 i 结尾的所有子数组的最大和就是以 i - 1 结尾的所有子数组的最大和 + nums[i]。
得到状态转移方程:
d
p
[
i
]
=
M
a
t
h
.
m
a
x
(
n
u
m
s
[
i
]
,
d
p
[
i
−
1
]
+
n
u
m
s
[
i
]
)
dp[i] = Math.max(nums[i], dp[i - 1] + nums[i])
dp[i]=Math.max(nums[i],dp[i−1]+nums[i]) 化简得
d
p
[
i
]
=
n
u
m
s
[
i
]
+
M
a
t
h
.
m
a
x
(
0
,
d
p
[
i
−
1
]
)
dp[i] = nums[i] + Math.max(0, dp[i - 1])
dp[i]=nums[i]+Math.max(0,dp[i−1])
题目要求子数组最少包含一个元素。若给定数组只有一个元素,那连续子数组的最大和就是该元素,则有初始化条件: d p [ 0 ] = n u m s [ 0 ] dp[0] = nums[0] dp[0]=nums[0]。
三、代码
class Solution {
public int maxSubArray(int[] nums) {
if(nums.length == 0)
return 0;
int[] dp = new int[nums.length];
dp[0] = nums[0];
int maxSum = dp[0];// 用该变量记录最大和,并返回
for(int i = 1; i < nums.length; i++){
dp[i] = nums[i] + Math.max(0, dp[i - 1]);
maxSum = Math.max(maxSum, dp[i]);// 更新最大和
}
return maxSum;
}
}