不停地筛选题,找了一个简单的数组题,看完题以为很简单就开始写了,然后发现。。。。还是自己会的太少了;
题目如下:
给定一个整数数组 nums ,找到一个具有最大和的连续子数组(子数组最少包含一个元素),返回其最大和。
示例:
输入: [-2,1,-3,4,-1,2,1,-5,4]
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大,为 6。
自己的算法及代码:
看完解析之后,我的思路是这样的:
- 从头遍历,之前 n 个数的和如果小于 0,就不算前面 n 个数,从第 n+1 个数开始算,即将前面的和置为 0;
- 每次求和之后,将和保存到新的数组里,然后求数组里的最大值,即为最大子序和;
public class Solution {
public static int maxSubArray(int[] nums){
int sum1 = 0;
int max = 0;
int[] sums = new int[nums.length];
for (int i = 0; i < nums.length; i++){
if (sum1 < 0){
sum1 = 0;
}
sum1 = sum1 + nums[i];
sums[i] = sum1;
}
//找最大值
max = sums[0];
for (int i = 0; i < nums.length; i++){
if (max < sums[i]){
max = sums[i];
}
}
return max;
}
public static void main(String[] args) {
int[] nums = new int[]{-2, 1, -3, 4, -1, 2, 1, -5, 4};
int sum = maxSubArray(nums);
System.out.println(sum);
}
}
参考的算法及代码:
这道题用到了 动态规划 的思想;
假设 nums 数组的长度是 n,下标从 0 到 n - 1。
我们用 ai 代表 nums[i],用 f(i) 代表以第 i 个数结尾的「连续子数组的最大和」,那么很显然我们要求的答案就是:
因此我们只需要求出每个位置的 f(i),然后返回 f 数组中的最大值即可。那么我们如何求 f(i) 呢?我们可以考虑 ai 单独成为一段还是加入 f(i - 1) 对应的那一段,这取决于 ai 和 f(i - 1) + ai 的大小,我们希望获得一个比较大的,于是可以写出这样的动态规划转移方程:
不难给出一个时间复杂度 O(n)、空间复杂度 O(n) 的实现,即用一个 f 数组来保存 f(i) 的值,用一个循环求出所有 f(i)。考虑到 f(i) 只和 f(i - 1) 相关,于是我们可以只用一个变量 pre 来维护对于当前 f(i)f(i) 的 f(i - 1) 的值是多少,从而让空间复杂度降低到 O(1),这有点类似「滚动数组」的思想。
public class Solution {
public static int maxSubArray(int[] nums){
int pre = 0, maxAns = nums[0];
for (int x : nums) {
pre = Math.max(pre + x, x);
maxAns = Math.max(maxAns, pre);
}
return maxAns;
}
}