LintCode解題筆記
阿新 • • 發佈:2019-01-24
給定一個整數陣列,請找出一個連續子陣列,使得該子陣列的和最大。輸出答案時,請分別返回第一個數字和最後一個數字的值。(如果兩個相同的答案,請返回其中任意一個)
樣例
給定 [-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;
}
};