HDU1024
题意:
把一个数组分成 m m m段,然后使得 s u m ( i 1 , j 1 ) + s u m ( i 2 , j 2 ) . . . . + s u m ( i m , j m ) sum(i1,j1)+sum(i2,j2).... + sum(im,jm) sum(i1,j1)+sum(i2,j2)....+sum(im,jm) 每对 ( i , j ) (i,j) (i,j)之间连续
思路:
d p [ i ] [ j ] dp[i][j] dp[i][j]代表在选择第 j j j个数字的时候,将前 j j j个数字分成 i i i组的最大和情况
这个时候 d p [ i ] [ j ] dp[i][j] dp[i][j]由两个部分的贡献组成
- ( x 1 , y 1 ) , ( x 2 , y 2 ) . . . ( x i , y i , a [ j ] ) (x1,y1) ,(x2,y2) ... (xi,yi,a[j]) (x1,y1),(x2,y2)...(xi,yi,a[j])
- ( x 1 , y 1 ) , ( x 2 , y 2 ) . . . ( x i − 1 , y i − 1 ) , ( a [ j ] ) (x1,y1) ,(x2,y2) ... (xi-1,yi-1),(a[j]) (x1,y1),(x2,y2)...(xi−1,yi−1),(a[j])
所以状态转移方程就是 : d p [ i ] [ j ] = m a x ( d p [ i ] [ j − 1 ] , d p [ i − 1 ] [ j ] ) + a [ j ] dp[i][j] = max(dp[i][j-1],dp[i-1][j])+a[j] dp[i][j]=max(dp[i][j−1],dp[i−1][j])+a[j]
然后我们发现时间复杂度和空间复杂都很不友好–
于是使用滚动数组来做进一步优化
我们注意到: d p [ i ] [ ∗ ] dp[i][*] dp[i][∗]只和 d [ i ] [ ∗ ] , d [ i − 1 ] [ ∗ ] d[i][*],d[i-1][*] d[i][∗],d[i−1][∗]有关,即当前状态只与前一状态有关
d p [ t ] [ j ] = m a x ( d p [ t ] [ j − 1 ] , d p [ 1 − t ] [ k ] ) + a [ j ] dp[t][j]= max(dp[t][j-1],dp[1-t][k])+a[j] dp[t][j]=max(dp[t][j−1],dp[1−t][k])+a[j] 其中 k = i − 1. i . . j − 1 , t = 1 k = i-1. i .. j-1, t = 1 k=i−1.i..j−1,t=1
只需要记录两个状态,当前状态为 t = 1 t = 1 t=1 下一状态为 t = 0 t = 0 t=0
然后现在我们只解决了空间的优化问题
优化时间:
我们不需要j-1之前的所有具体的k在哪个位置取到最大值,所有我们只需要使用一个pre数组来存在当前点前的最大值即可
则转移方程变为
d p [ t ] [ j ] = m a x ( d p [ t ] [ j − 1 ] , p r e [ j − 1 ] ) + a [ j ] dp[t][j]=max(dp[t][j-1],pre[j-1])+a[j] dp[t][j]=max(dp[t][j−1],pre[j−1])+a[j]
这个时候可以去掉第一维,变成:
d p [ j ] = m a x ( d p [ j − 1 ] , p r e [ j − 1 ] ) + a [ j ] dp[j]=max(dp[j-1],pre[j-1])+a[j] dp[j]=max(dp[j−1],pre[j−1])+a[j]
Code
#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
int s[N];
int pre[N];
int n,m;
int dp[N];
int main()
{
while(~scanf("%d%d",&m,&n))
{
for(int i = 1;i <= n;i++)scanf("%d",&s[i]);
pre[0] = 0;
int tmp;
for(int j = 1;j <= m;j++)
{
tmp = -0x3f3f3f3f;
for(int i = j;i <= n;i++)
{
dp[i] = max(dp[i-1],pre[i-1]) + s[i];
pre[i-1] = tmp;
tmp = max(tmp,dp[i]);
}
}
cout << tmp <<"\n";
}
return 0;
}