HDU 2746 ——Max Sum Plus Plus Plus 【dfs+带备忘 或 DP】

来源:互联网 发布:新手怎样做淘宝客 编辑:程序博客网 时间:2024/06/05 07:02
题目:http://acm.hdu.edu.cn/showproblem.php?pid=1244

Problem Description
给定一个由n个正整数组成的整数序列

a1 a2 a3 ... an

求按先后次序在其中取m段长度分别为l1、l2、l3...lm的不交叠的连续整数的和的最大值。
 
Input
第一行是一个整数n(0 ≤ n ≤ 1000),n = 0表示输入结束
第二行的第一个数是m(1 ≤ m ≤ 20),
第二行接下来有m个整数l1,l2...lm。
第三行是n个整数a1, a2, a2 ... an.
 
Output
输出m段整数和的最大值。
 
Sample Input
32 1 11 2 342 1 21 2 3 50
 
Sample Output
5 10

 

这道题最好的办法的DP,然而我第一直觉是:DFS + 记忆化,有点接近模拟的方式去找全部的情况,一次就AC了 :)

方法1:DFS + 记忆化

#include <cstdio>#include <iostream>#include <cstring>#include <string>#define INF 0x3f3f3f3fusing namespace std;int a[1005];int sum[1005];int l[25];int dp[25][1005];int n, m;int dfs(int i, int cur){        if(cur > n)    return -INF;        if(i==m+1)    return 0;        if(dp[i][cur])    return dp[i][cur];        int ret=-INF;    for(int k=cur; k<n; k++) {        ret = max(ret, dfs(i+1, k+l[i])+sum[k+l[i]]-sum[k]);    }    return dp[i][cur] = ret;}int main (){    while(scanf("%d", &n) != EOF && n) {        scanf("%d", &m);        for(int i=1; i<=m; i++) {            scanf("%d", &l[i]);        }        for(int i=1; i<=n; i++) {            scanf("%d", &a[i]);            sum[i] = sum[i-1] + a[i];        }        printf("%d\n", dfs(1, 0));        memset(dp, 0, sizeof(dp));    }            return 0;}

方法2:DP

定义:

  dp[i][j]:区间[1,j]的序列中,i个连续子序列的和的最大值  (注意这里的[1,j]是一个松的边界,i个连续子序列藏身于其中)

  sum[i]:a[1]+a[2]+...+a[i]  前缀和

则:

  dp[i][j] = max(dp[i][j-1] (a[j]不属于最后一个子序列) , dp[i-1][j-l[i]]+sum[j]-sum[j-l[i]] (a[j]属于最后一个子序列) )

#include <cstdio>#include <iostream>#include <cstring>#include <string>#define INF 0x3f3f3f3fusing namespace std;int a[1005];int sum[1005];int l[25];int dp[25][1005];int n, m;int main (){    while(scanf("%d", &n) != EOF && n) {        scanf("%d", &m);        for(int i=1; i<=m; i++) {            scanf("%d", &l[i]);        }        for(int i=1; i<=n; i++) {            scanf("%d", &a[i]);            sum[i] = sum[i-1] + a[i];        }        int cur = 0;        for(int i=1; i<=m; i++) {            cur += l[i];            for(int j=cur; j<=n; j++) {                dp[i][j] = dp[i][j-1];                dp[i][j] = max(dp[i][j], dp[i-1][j-l[i]] + sum[j] - sum[j-l[i]]);            }        }        printf("%d\n", dp[m][n]);    }            return 0;}

 DP 的进一步空间优化

  利用滚动数组的原理,将原来的O(mn)压缩为O(n)。但是,仔细观察会发现,将空间压缩为一个一维数组会很麻烦,退而求其次,采用两个一维数组的空间:
#include <cstdio>#include <iostream>#include <cstring>#include <string>#define INF 0x3f3f3f3fusing namespace std;int a[1005];int sum[1005];int l[25];int dp[2][1005];int n, m;int main (){    while(scanf("%d", &n) != EOF && n) {        scanf("%d", &m);        for(int i=1; i<=m; i++) {            scanf("%d", &l[i]);        }        for(int i=1; i<=n; i++) {            scanf("%d", &a[i]);            sum[i] = sum[i-1] + a[i];        }        int cur = 0;        for(int i=1; i<=m; i++) {            cur += l[i];            for(int j=cur; j<=n; j++) {                dp[i%2][j] = dp[i%2][j-1];                dp[i%2][j] = max(dp[i%2][j], dp[1-i%2][j-l[i]] + sum[j] - sum[j-l[i]]);            }        }        printf("%d\n", dp[m%2][n]);        memset(dp, 0, sizeof(dp)); // 滚动数组优化后,记得加多一个清0     }            return 0;}

 

0 0
原创粉丝点击