开始拿到这题,想法是使用前缀和数组去完成,但是在寻找最大值的过程中使用了两重的for循环,时间复杂度达到了n^2,再遇到较大的样例时,超时了。先贴一下自己的代码:
class Solution {
public:
int maxSubArray(vector<int>& nums) {
int N = nums.size();
int maxx = -(1e5);
int s[N + 10];
if(N == 1) {
maxx = nums[0];
}else {
for(int i=1;i<=N;i++) {
s[i] = s[i-1] + nums[i-1];
}
for(int i=0;i<N;i++) {
for(int j=i+1;j<=N;j++) {
if((s[j]-s[i]) > maxx) {
maxx = s[j] - s[i];
}
}
}
}
return maxx;
}
};
之后换了一种思路,既然for两次会超时,那我就for一次,时间复杂度就是O(n)的了。修改的主要思想就是贪心,取局部最优解。假设左右两个数字,左边的数字已经小于0时,无论右边数字的正负,二者相加,数值一定会更小,此时我就舍弃左边的数字,从右边的数字开始计算。在计算的过程中,使用一个变量保存下最大值,sum<0 就可以理解为舍弃左边数字的这个过程,最后return即可。AC代码如下:
class Solution {
public:
int maxSubArray(vector<int>& nums) {
int sum = 0, maxx = -(1e5);
if(nums.size() == 1) {
maxx = nums[0];
}else {
for(int i=0;i<nums.size();i++) {
sum += nums[i];
if(sum > maxx) {
maxx = sum;
}
if(sum < 0) {
sum = 0;
}
}
}
return maxx;
}
};