dp背包训练

1. 01背包

最基础的背包问题,要求熟练掌握空间优化的方法.

hdu 2602 Bone Collector

#include <bits/stdc++.h>
using namespace std;
int t, n, v, c[1005], w[1005], dp[1005];

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

hdu 2955 Robberies

说有一个小偷n家银行,偷第i家能获得Mi的钱,同时被抓的概率是Pi,问你怎么可以偷到最多的钱同时被抓的总概率不超过Pj.

---

初始考虑时,会想到这么一个状态dp[i, j]表示前i家被抓的概率在j之下可获得的最大利润。但是Pj是一个浮点数,因此并不能这样来定义状态,于是稍微转换一下,定义dp[i, j]为前i家银行，偷钱为j时的最大不被抓概率，那么递推如下：

dp[i, j] = max(dp[i-1][j], dp[i-1][j-Mi]*(1-Pi));
关键点在于乘法公式,其实是高中的概率问题
比如偷三家的被抓的概率分别是p1,p2,p3，那么不被抓的概率是1-(1-p1)(1-p2)(1-p3)
因为题目要求是总的被抓的概率要小于等于P,而这个被抓不如不被抓来得好考虑一些。

---

#include <bits/stdc++.h>
using namespace std;

int t, n, m[105];
double P, p[105], dp[10005];

int main() {
    scanf("%d", &t);
    while (t--) {
        memset(dp, 0, sizeof dp);
        scanf("%lf %d", &P, &n);
        int v = 0;
        for (int i = 0; i < n; i++) {
            scanf("%d %lf", &m[i], &p[i]);
            v += m[i];
        }
        dp[0] = 1;
        for (int i = 0; i < n; i++) 
            for (int j = v; j >= m[i]; j--) {
                dp[j] = max(dp[j], dp[j-m[i]]*(1-p[i]));
            }

        while (dp[v] < 1-P) v--;
        printf("%d\n", v);  
    } 
    return 0;
}

CodeForces 366C Dima and Salad

有n种水果,每种水果有两个指标ai,bi，ai指水果的美味度，bi代表水果的卡路里。现在问你怎么样从这些水果中挑选，可以满足$\frac{\sum a_i}{\sum b_i} = k$,其中k为给定的常数且使得选中水果的美味度最大。

---

其实这是一个背包问题，作如下替换

a[i] convert to a[i]-k*b[i]

那么问题就变为一堆物品,其容量为a[i]-k*b[i],现在将这些物品装满一个容量为0的背包，问能否装满，如果能装满，那么题目中求的水果的美味度最大又是多少。

---

抽象成一个数学问题,给一个数组，有正有负，问能否从中挑选出几个数，使其和为0？
由于容量有正有负,所以并不好处理。本题这里将两者分开处理，分别做一次背包，最后得到结果

---

#include <bits/stdc++.h>
using namespace std;
const int inf = 0x3f3f3f3f;

int n, k, a[105], b[105], c[105], dp[10005], dp2[10005];

int main() {
    scanf("%d %d", &n, &k);
    for (int i = 0; i < n; i++)
        scanf("%d", &a[i]);
    for (int i = 0; i < n; i++) {
        scanf("%d", &b[i]);
        c[i] = a[i] - k*b[i];
    }

    for (int i = 0; i <= 10000; i++)
        dp[i] = dp2[i] = -inf;
    dp[0] = dp2[0] = 0;

    for (int i = 0; i < n; i++) {
        if (c[i] > 0) {
            for (int j = 10000; j >= c[i]; j--)
                dp[j] = max(dp[j], dp[j-c[i]]+a[i]);
        } else {
            c[i] = -c[i];
            for (int j = 10000; j >= c[i]; j--)
                dp2[j] = max(dp2[j], dp2[j-c[i]]+a[i]);
        }
    }
    int ans = -1;
    for (int i = 0; i < 10000; i++) {
        if (dp[i] > 0 && dp2[i] > 0)
            ans = max(ans, dp[i]+dp2[i]);
    }
    printf("%d\n", ans);
    return 0;
}

hdu 1114 Piggy-Bank

有一个储蓄罐，能装n种硬币，第i种硬币价值为wi，重量为pi，现在已知该储蓄罐重量，问你用所给的硬币装进储蓄罐能不能达到该重量，如果可以，装进该储蓄罐的这些硬币的最小价值是多少。

---

完全背包问题。完全背包问题和01背包问题相比，只是在第二层循环的顺序上有区别。注意初始化条件，题目要求是装满，所以只有dp[0] = 0,又由于求解的是最小价值，因此其余状态应该定义为+oo。

---

#include <bits/stdc++.h>
using namespace std;
const int oo = 0x3f3f3f3f;

int t, e, f, n, p[505], w[505], dp[10005];
int main() {
    scanf("%d", &t);
    while (t--) {
        scanf("%d %d %d", &e, &f, &n);
        for (int i = 0; i < n; i++)
            scanf("%d %d", &p[i], &w[i]);
        f -= e;
        for (int i = 1; i <= f; i++) dp[i] = oo;
        dp[0] = 0;
        for (int i = 0; i < n; i++)
            for (int j = w[i]; j <= f; j++)
                dp[j] = min(dp[j], dp[j-w[i]]+p[i]);
        if (dp[f] == oo)
            printf("This is impossible.\n");
        else
            printf("The minimum amount of money in the piggy-bank is %d.\n", dp[f]); 
    }
    return 0;
} 

hdu 5534 Partial Tree

有一个图，图初始只有n个节点，现在往该图上添加n-1条边，要求最后图中任意两点连通。现在对每一个节点定义一个得分函数f(i)，其中i是该节点的度。现在问你怎么添加可以既保证连通又保证所有节点的得分和最大。

---

首先要保证连通，其实这怎么做呢？其实可以从度的角度去考虑，也就是说我不去考虑这棵树的结构是怎么样的，是怎么样一根一根加的，我只考虑最后每个节点的度怎么样就行。添加n-1条边，每添加一条边都会增加2个度，因此该图的总度数为2(n-1)。现在，我首先先给每一个点分配一个度(其实这就可以保证题目的连通性，然后剩下n-2个度，分配给n个点，问怎么样分配可以得到最大价值。这显然就是一个完全背包的问题了。

---

#include <bits/stdc++.h>
using namespace std;

const int oo = 0x3f3f3f3f;

int t, n, f[2018], dp[2018];
int main() {
    scanf("%d", &t);
    while (t--) {
        scanf("%d", &n);
        for (int i = 1; i < n; i++)
            scanf("%d", &f[i]);
        for (int i = 0; i < n; i++)
            dp[i] = -oo;
        dp[0] = 0;
        for (int i = 1; i < n-1; i++) {
            for (int j = i; j <= n-2; j++) {
                dp[j] = max(dp[j], dp[j-i]+f[i+1]-f[1]);
            }
        }
        printf("%d\n", dp[n-2]+n*f[1]);
    }
    return 0;
}