原题地址: http://www.lintcode.com/zh-cn/problem/continuous-subarray-sum/
给定一个整数数组,请找出一个连续子数组,使得该子数组的和最大。输出答案时,请分别返回第一个数字和最后一个数字的值。(如果两个相同的答案,请返回其中任意一个)
样例
给定 [-3, 1, 3, -3, 4], 返回[1,4].
分析:
使用start和end来定位A中最大子数组开始和结束的位置
方法一: 暴力遍历的方式,两层for循环定位到i,j,然后求数组A[i..j]的和,求和的复杂度是O(n), 两层遍历的时间复杂度是O(n*n), 则一共遍历复杂度是O(n*n*n)
方法二:采用DP动态规划方式,假设sum[i],是前i项的最大值max,
如果sum[i] >0,判断A[i+1]的符号,如果A[i+1]<0, 则A[i+1]无法增大max的值,因此sum[i+1]还是max,start和end的标识不变。如果A[i+1]>0,则A[i+1]对总和最大有正向贡献,因此sum[i+1] = sum[i] + A[i+1], 同时end指向i+1
如果sum[i]<0, 说明前i项对总和是负向贡献,因此需要重新开辟求和的位置,sum[i+1]舍弃前i项的和,直接从A[i+1]开始,sum[i+1] = A[i+1], start and end 重新定位到i+1
循环,直到遍历A结束,那么复杂度就是O(n)
class Solution {
public:
/**
* @param A an integer array
* @return A list of integers includes the index of
* the first number and the index of the last number
*/
vector<int> continuousSubarraySum(vector<int>& A) {
// Write your code here
if(A.size() < 1){
return A;
}
int start=0;
int end=0;
long max;
long current;
vector<int> vect;
int sum[A.size()];
//init
vect.clear();
vect.push_back(0);
vect.push_back(0);
for(int i = 0; i<A.size(); i++){
sum[i] = 0;
}
sum[0]=A[0];
int f_start = 0;
int f_end = 0;
for(int i=1, max=A[0]; i< A.size(); i++){
if(sum[i-1]>0){
sum[i] = sum[i-1]+A[i];
end = i;
}else{
sum[i] = A[i];
start = i;
end = i;
}
//max = (max > sum[i])? max:sum[i];
if(max < sum[i]){
max = sum[i];
f_start = start;
f_end = end;
}
}
if(f_start!=0 || f_end!=0){
vect.clear();
vect.push_back(f_start);
vect.push_back(f_end);
}
return vect;
}
};