面试算法

子数组最大累加和

2018-09-26  本文已影响1306人  wenyilab

题目:子数组最大累加和

给定一个数组arr,返回子数组的最大累加和

例如:arr = [1,-2,3,5,-2,6,-1],所有子数组中[3,5,-2,6]可以累加出最大的和12,所以返回12.

【要求】arr的长度为N,时间复杂度为O(N),额外空间复杂度为O(1)

【思路】用变量cur记录每一步的累加和,遍历到正数cur增加,遍历到负数cur减小。当cur<0时,说明累加到当前数出现了小于0的结果,那么累加的这一部分肯定不能作为产生最大累加和的子数组的左边部分,此时令cur = 0,表示从下一个数开始累加;当cur>0,每一次累加都可能是最大的累加和。用max跟踪记录cur出现的最大值。

代码实现 python
def maxSumSubArr(arr,n):
    if (arr == null || n < 1):
        return 0
    res = arr[0]
    cur = 0
    for i in range(n):
        cur += arr[i]
        res = max(res,cur)
        cur = cur < 0 ? 0 : cur
    return res
代码实现 c++
#include <iostream>
#include <climits> //C++中的头文件 

using namespace std; 

int maxSumSubArr(int *arr, int n){    
    if (arr == NULL || n < 1)        
      return 0;    
    int res = INT_MIN; //结果   
    int cur = 0; //当前和        
    for (int i = 0; i < n; i++){        
      cur += arr[i]; //直接依次加上       
      res = max(res, cur);       
      cur = cur < 0 ? 0 : cur;    
  }    
  return res;
}

int main(){   
    int arr[] = {-2, 1, 3, -2, 1, -2, 1};   
    cout << maxSumSubArr(arr, 7);    
    return 0;      
}
上一篇下一篇

猜你喜欢

热点阅读