洛谷 P1115 最大子段之和 线性DP

这道题目看到很多大佬写的很复杂，其实代码真的好简单，一开始也被自己绕复杂了，用了前缀和数组，也自然用了i,j两个维度,i当做区间末端，j当做区间头，但是这样o(n*n)的时间复杂度，于是降维，把j那个维度去掉，降了就AC了

先看题目

题目描述

给出一个长度为 nn 的序列 aa，选出其中连续且非空的一段使得这段和最大。

输入格式

第一行是一个整数，表示序列的长度 nn。

第二行有 nn 个整数，第 ii 个整数表示序列的第 ii 个数字 a_iai​。

输出格式

输出一行一个整数表示答案。

输入输出样例

输入 #1复制

7
2 -4 3 -1 2 -4 3

输出 #1复制

4

说明/提示

样例 1 解释

选取 [3, 5][3,5] 子段 \{3, -1, 2\}{3,−1,2}，其和为 44。

数据规模与约定

• 对于 40\%40% 的数据，保证 n \leq 2 \times 10^3n≤2×103。
• 对于 100\%100% 的数据，保证 1 \leq n \leq 2 \times 10^51≤n≤2×105，-10^4 \leq a_i \leq 10^4−104≤ai​≤104。

 先上o(n*n)超时代码

#include <iostream>
#include <cmath>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
using namespace std;
int n;
const int N=1e5+10;
int a[N];
int s[N];
int dp[N];
int main()
{
    cin>>n;
    bool flag=false;
    int res=-0x3f3f3f3f;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        s[i]=s[i-1]+a[i];       //前缀和数组
        for(int j=0;j<i;j++)   //超时所在
        {
            dp[i]=max(a[i],dp[j]+s[i]-s[j]);
            res=max(res,dp[i]);
        }
    }
    cout<<res<<endl;
}

 下面是优化后的代码

#include <iostream>
#include <cmath>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
using namespace std;
int n;
const int N=1e6+10;
int a[N];
int dp[N];
int main()
{
    cin>>n;
    int res=-0x3f3f3f3f;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
                      //s[i]=s[i-1]+a[i];
                     //dp[i]=max(a[i],dp[i-1]+s[i]-s[i-1]);
        dp[i]=max(a[i],dp[i-1]+a[i]);
        res=max(res,dp[i]);
    }
    cout<<res<<endl;
}