1. 程式人生 > >LintCode解題筆記

LintCode解題筆記

給定一個整數陣列,請找出一個連續子陣列,使得該子陣列的和最大。輸出答案時,請分別返回第一個數字和最後一個數字的值。(如果兩個相同的答案,請返回其中任意一個)

樣例
給定 [-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; } };