动态规划——背包问题

一、01背包
题目描述:01背包是在n件物品取出若干件放在空间为W的背包里,每件物品的体积为w1,w2……wn,与之相对应的价值为P1,P2……Pn。求出获得最大价值的方案。[注意:每种物品只能拿一个。]

用二位数组dp来记录状态,其中dp[i][j]表示的是从第一件物品开始向背包里装,当背包最大空间为j且装到第i件物品时,价值最大是多少。显然,对于每个物品可以选择装或不装~
递推方程:

dp[0][j]=0;
dp[i+1][j]=dp[i][j](j<w[i])
dp[i+1][j]=max(dp[i][j],dp[i][j-w[i]]+p[i+1])(j>=w[i])

例如数据为(w,p)={(2,3),(1,2),(3,4),(2,3)}
根据递推公式可得下表:

这里写图片描述

实现代码如下:

memset(dp, 0, sizeof(dp));
for (int i = 1; i <= n; i++)
{
    for (int j = 0; j <= W; j++)
    {
        if (w[i] > j)
            dp[i][j] = dp[i - 1][j];
        else
        {
            dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - w[i]] + p[i]);
        }
    }
}

  

时间复杂度O(nw)

也可以优化成一维数组,代码如下

for (int i = 1; i <= n; i++)
{
    for (int j = W; j >= w[i]; j--)//这里需要注意,j从W开始循环,因为会用到 dp[j - w[i]],从0到W会出错~~
    {
        dp[j] = max(dp[j], dp[j - w[i]] + p[i]);
    }
}

  

例题Hdu Problem : 2602 ( Bone Collector )

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAX = 1005;
int d[MAX], w[MAX], p[MAX];

int main()
{
    int n, W, t;

    scanf("%d", &t);
    while (t--)
    {
        scanf("%d%d", &n, &W);
        for (int i = 1; i <= n; i++)
            scanf("%d", &p[i]);
        for (int i = 1; i <= n; i++)
            scanf("%d", &w[i]);
        memset(d, 0, sizeof(d));
        for (int i = 1; i <= n; i++)
            for (int j = W; j >= w[i]; j--)
                    d[j] = max(d[j], d[j - w[i]] + p[i]);
        printf("%d\n", d[W]);
    }
    return 0;
}

  

二、完全背包
与01背包的区别就是一种物品可以有无数多件。所以对于每种物品,可以选择放0件,放1件,放n件……在不超过背包重量的情况下……
递推公式(复杂度O(nW^2))

dp[0][j]=0;
dp[i][j]=max(dp[i-1][j-k*w[i]]+k*p[i]);(k=0,1,2,...)(j>k*w[i])

通过观察可发现dp[i][j]可以由dp[i][j-w[i]]推出。从而降低时间复杂度

实现代码(时间复杂度O(nW)):

memset(dp, 0, sizeof(dp));
for (int i = 0; i < n; i++)
{
    for (int j = 0; j <= W; j++)
    {
        if (w[i] > j)
            dp[i + 1][j] = dp[i][j];
        else
            dp[i + 1][j] = max(dp[i][j], dp[i + 1][j - w[i]] + p[i]);
    }
} 

一维数组实现

for (int i = 0; i < n; ++i) {
	for (int j = w[i]; j <= W; ++j) {
		dp[j] = max(dp[j], dp[j - w[i]] + p[i]);
	}
}
printf("%d\n", dp[W]);

  

posted @ 2015-03-17 22:30  我不吃饼干呀  阅读(186)  评论(0编辑  收藏  举报