参考资料
本题tyvj题解
《算法艺术与信息学竞赛》
Tyvj P1305 最大子序和
时间: 1000ms / 空间: 131072KiB / Java类名: Main
描述
输入一个长度为n的整数序列,从中找出一段不超过M的连续子序列,使得整个序列的和最大。
例如
1,-3,5,1,-2,3
当m=4时,S=5+1-2+3=7
当m=2或m=3时,S=5+1=6
输入格式
第一行两个数n,m
第二行有n个数,要求在n个数找到最大子序和
输出格式
一个数,数出他们的最大子序和
测试样例
输入
6 4
1 -3 5 1 -2 3
输出
7
备注
数据范围:
100%满足n,m<=300000
第一步:50分
这是经典问题最大子数组的修改版,乍一看似乎有点无从下手。但是经过分析不难得出一个2D/0D的dp方程。不妨用f(i,j)表示前i个数向前j个的子序和的最大值。注意题目中的要求 不超过M ,所以决策有以下几种:
- 当j=0时,f(i,j) = 0
- 如果f(i-1, j-1) > 0,那么f(i, j) = f(i-1, j-1)+a[i]
- 否则 f(i,j) = 0
在这个思想的指导下,不难写出dp方程:
f(i,j) = max(0, f(i-1, j-1)+a[i])
最后的结果是max{f(i,M) | 1≤i≤n}
显然i,j都是递增的,只要简单地i,j递增dp就可以解决问题了
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n, m;
int a[4001];
int dp[4001][4001];