题意:
给n个数值, 求其中m段连续的子串之和最大。
状态转移方程(复杂度:n * n * m):
- f [ i ][ j ] = max (f [ i - 1 ][ j ] , max (f [ k ][ j - 1 ])); (k = j - 1,…… i - 2,i - 1);(状态转移:O(n))
- f [ i ][ j ] :前 i 个数值划分出 j 段子串(第 i 个数值一定在子串的最后一个位置)的和的最大值。
- 结果:max ( f [ k ][ m ]); (k = m,…… n - 1,n)
原理:
第i个数值,要不在前面的子串后面(子串数目不变),要不自己组成一个新的子串。
优化:
引入一个辅助数组g [ i ][ j ](前 i 个数值划分出 j 段子串的和的最大值)。
状态转移方程(复杂度:n * m):
f [ i ][ j ] = max (f [ i - 1 ][ j ], g [ i - 1 ][ j - 1 ]); (状态转移:O(1))
g [ i ][ j ] = max (f [ i ][ j ] , g [ i - 1 ][ j ]);
写程序的时候可以用滚动数组来解决问题。
学成因为数组有两种方法:
一:
#include <cstdio> #include <cstring> #include <algorithm> using namespace std; const int INF = 0x7fffffff, N = 1000000; int f[N], g[N], num[N]; int SumPP (int n, int m) { int i, j, t; for (j = 1; j <= m; j++) { f[j - 1] = -INF; for(i = j; i <= n; i++) { f[i] = max(f[i - 1], g[i - 1]) + num[i]; if(i < j + 2) { g[i - 1] = (i == j)? -INF : f[i - 1]; continue; } else { g[i - 1] = max(g[i - 2], f[i - 1]); } } g[i - 1] = max(g[i - 2], f[i - 1]); } return g[n]; } int main () { int n, m; while (scanf ("%d %d", &m, &n) != EOF) { g[0] = 0; for (int i = 1; i <= n; i++) { scanf("%d", &num[i]); f[i] = -INF; g[i] = 0; } printf("%d\n", SumPP(n, m)); } return 0; }
二:
#include <cstdlib> #include <cstdio> #include <cstring> #include <cmath> #include <algorithm> using namespace std; int num[1000000], n, m; int f[1000000], g[1000000]; int SumPP() { int i, j; f[1] = 0; for (i = 1; i <= n; i++) { int t = min(i, m); for(j = 1; j <= t; j++) { f[j] = max(f[j], g[j - 1]) + num[i]; g[j - 1] = max(f[j - 1], g[j - 1]); } g[j - 1] = max(f[j - 1], g[j - 1]); } return g[m]; } int main () { while (scanf ("%d %d", &m, &n) != EOF) { g[0] = -0x7fffffff; for (int i = 1; i <= n; i++) { f[i] = g[i] = -0x7fffffff; scanf ("%d", &num[i]); } printf("%d\n", SumPP()); } return 0; }