HDOJ 1003 Max Sum （dp思路详解）

原题链接：http://acm.hdu.edu.cn/showproblem.php?pid=1003

部分题头：

Problem Description

Given a sequence a[1],a[2],a[3]......a[n], your job is to calculate the max sum of a sub-sequence. For example, given (6,-1,5,4,-7), the max sum in this sequence is 6 + (-1) + 5 + 4 = 14.

 

Input

The first line of the input contains an integer T(1<=T<=20) which means the number of test cases. Then T lines follow, each line starts with a number N(1<=N<=100000), then N integers followed(all the integers are between -1000 and 1000).

不管什么题，第一想到的就是暴力枚举，用两重for枚举所有连续子序列，在内层for里加和并判断是否是最大和，是的话记录起始和结束下标（要注意，虽然题目说是sub sequence，但Output要求是起始下标和终止下标，说的是连续子序列），时间复杂度为 ，数据范围 1<=N<=100000，肯定超时。

毫无疑问，那就只能用dp了，首先定义dp[i]为以a[i]结尾的最大子序列和，以dp[3]为例，dp[3]就等于以a[3]结尾的子序列a[1] a[2] a[3]、a[2] a[3]、a[3] 里最大的子序列的和。dp[1] 就等于a[1]，因为以a[1]结尾的最大子序列和就是他本身。转移方程为dp[i] = max(dp[i-1] + a[i] , a[i]) ，这个转移方程个人宏观的理解是：当一段连续子序列的最大和（指的是公式里的dp[i-1]）为负数的时候，那么我们就舍弃这段子序列，从a[i]开始计算一段新的子序列（很显然dp[i-1] + a[i]都小于a[i]的话只能说明dp[i-1]是负数，那么对于要求最大子序列和，dp[i-1]是没有任何贡献的）。

以数组a = {1 ,2 ,-1 ,-2}为例，这里演示整个过程：

首先定义dp[1] = a[1]（要注意，数组a和数组dp都从1开始，0下标位不使用，故开数组的时候要多开一个空间不然会漏数据），这时候也要记录一下当前最大的子序列和maxsum和该子序列的起始和结束位置，然后从dp[2]开始递推，根据转移方程，先看dp[1]，dp[1] + a[2] = 3，而a[2] = 2，所以dp[2] = dp[2-1] + a[2]，即是dp[2] = 3，此时判断一下dp[2]是否大于maxsum，是，那就更新maxsum = dp[2]，然后记录一下最大和的子序列的起始和结束。然后继续计算dp[3]，dp[2] + a[3] > a[3]，所以dp[3] = dp[2] + a[3]，此时再判断dp[3]是否大于maxsum，否，那就不管，继续。计算dp[4]，dp[3] + a[4] > a[4]，所以dp[4] = dp[3] + a[4]，此时再判断一下maxsum是否小于dp[4]，否，不管，结束。最后的输出就是：最大子序列和为3，该子序列起始位置为1，结束位置为2。

我的ac代码：

#include <cstdio>
#include <cmath>
#include <algorithm>

using namespace std;

int main()
{

    int T;
    scanf("%d", &T);
    for (int i = 0; i < T; ++i)
    {
        int N;
        scanf("%d", &N);
        int *arr = new int[N + 1];
        for (int j = 1; j <= N; ++j)
            scanf("%d", &arr[j]);
        int *dp = new int[N + 1];
        int si = 1, ei = 1, maxsum = arr[1], csi = 1, cei = 1;
        //si = startIndex,ei = endIndex,si ei实时记录当前的序列起始和中止，csi cei记录更新maxsum时的最大和子序列的起始和终止，给最后输出答案用
        dp[0] = 0;
        dp[1] = arr[1];
        for (int j = 2; j < N + 1; ++j)
        {
            if (dp[j - 1] < 0)
            {
                dp[j] = arr[j];
                si = j;
                ei = j;
            }
            else
            {
                dp[j] = dp[j - 1] + arr[j];
                ei = j;
            }
            if (dp[j] > maxsum)
            {
                csi = si;
                cei = ei;
                maxsum = dp[j];
            }
        }
        printf("Case %d:\n%d %d %d\n", i + 1, maxsum, csi, cei);
        if (i != T - 1)
            printf("\n");

        delete[] arr;
        delete[] dp;
    }

    return 0;
}

后记

这是我第一次学习dp时做的题，其实回过来想想，使用dp解题的思路是 首先考虑子问题是什么，这题的子问题就是以a[i]结尾的子序列里的最大和子序列，然后根据这个子问题定义dp[i]，还要考虑初始定义dp[1] = a[1]，因为a[1]的最大和子序列只能是他本身，最后思考如何递推，也就是写出正确的转移方程，这里的逻辑是舍弃最大和为负的那段子序列，以保证所求的是最大和子序列，编码的时候还要注意题目要求起始位置和结束位置。