算法提高之动态规划：背包模型四

1、货币系统(完全背包问题求方案数)

#include <iostream>

using namespace std;

//存储的数据比较多  爆int
//如何判断long long 提交后报数组越界 考虑long long
typedef long long LL;

const int M = 3010;

int n, m;
LL f[M];

int main()
{
    cin >> n >> m;
	//一个物品不选，有一种方案
    f[0] = 1;
    for (int i = 0; i < n; i ++ )
    {
        int v;
        cin >> v;
        for (int j = v; j <= m; j ++ )
            f[j] += f[j - v];
    }

    cout << f[m] << endl;

    return 0;
}

2、货币系统

#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

//N是数据个数 M是最大数的数值
const int N = 25010,M=110;

int n;
int a[M];
int f[N];

int main()
{
    int T;
    cin >> T;
    while (T -- )
    {
        cin >> n;
        for (int i = 0; i < n; i ++ ) cin >> a[i];
        //排序是从a[0]开始的，因此输入的下标需要从0开始
        sort(a, a + n);
		//最大值 n从0->n-1 则从小到大排序后，a[n-1]为最大值
        int m = a[n - 1];
        //有T组数据，需要对上一组数据更新
        memset(f, 0, sizeof f);
        //状态方程定义的是方案数，表示一个不选的时候有一种方案
        f[0] = 1;

        int res = 0;
        for (int i = 0; i < n; i ++ )
        {
        	//第一次执行 被选择的集合中为空，因此一定选择
        	//第二次被a[0]更新了一遍状态方程,a[1]能被表示，则方案数不为0
            if (!f[a[i]]) res ++ ;
            //一维完全背包，从小到大枚举
            for (int j = a[i]; j <= m; j ++ )
                f[j] += f[j - a[i]];
        }

        cout << res << endl;
    }

    return 0;
}

3、混合背包问题（多重背包使用二进制优化）

#include <iostream>

using namespace std;

const int N = 1010;

int n, m;
int f[N];

int main()
{
    cin >> n >> m;

    for (int i = 0; i < n; i ++ )
    {
        int v, w, s;
        cin >> v >> w >> s;
        if (!s)//完全背包
        {	
        //从小到大循环
            for (int j = v; j <= m; j ++ )
                f[j] = max(f[j], f[j - v] + w);
        }
        else
        {
        //把01背包看成特殊的完全背包
            if (s == -1) s = 1;
        //边二级制优化 边转换成01背包解题
            for (int k = 1; k <= s; k *= 2)
            {
                for (int j = m; j >= k * v; j -- )
                    f[j] = max(f[j], f[j - k * v] + k * w);
                s -= k;
            }
            if (s)
            {
                for (int j = m; j >= s * v; j -- )
                    f[j] = max(f[j], f[j - s * v] + s * w);
            }
        }
    }

    cout << f[m] << endl;

    return 0;
}

4、有依赖的背包问题(分组背包与树形dp)

#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 110;

int n, m;
int v[N], w[N];
int h[N], e[N], ne[N], idx;
int f[N][N];

void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}

void dfs(int u)
{
	//找到以u为根节点的所有子树  将所有子树遍历完后再去添加u物品->更新状态
    for (int i = h[u]; i!=-1; i = ne[i])   // 循环物品组
    {
    //子节点的编号
        int son = e[i];
        dfs(e[i]);

        // 分组背包 体积从大到小循环 由于必须加上u这个点 预先留出位置
        for (int j = m - v[u]; j >= 0; j -- )  // 循环体积
            for (int k = 0; k <= j; k ++ )  // 循环决策 当前使用多少体积
            	//f[son][k] 给子节点k个体积能返回多少价值
                f[u][j] = max(f[u][j], f[u][j - k] + f[son][k]);
    }

    // 将物品u加进去
    for (int i = m; i >= v[u]; i -- ) f[u][i] = f[u][i - v[u]] + w[u];
    //放不进去u的位置 写成0
    for (int i = 0; i < v[u]; i ++ ) f[u][i] = 0;
}

int main()
{
    cin >> n >> m;
	//清空邻接表头节点
    memset(h, -1, sizeof h);
    int root;
    for (int i = 1; i <= n; i ++ )
    {
        int p;
        cin >> v[i] >> w[i] >> p;
        if (p == -1) root = i;
        else add(p, i);
    }

    dfs(root);
	//以root为根节点 体积不超过m的最大价值
    cout << f[root][m] << endl;

    return 0;
}

5、背包问题求方案数（最优解方案数）

#include <cstring>
#include <iostream>

using namespace std;

const int N = 1010, mod = 1e9 + 7;

int n, m;
int f[N], g[N];

int main()
{
    cin >> n >> m;
	/*状态方程 f[i,j] 恰好是j的选法。因此初始化的时候f[0]=0 其余状态等于
	无穷大，可以参考潜水员。由于是最大值，所以使用负无穷。对于最小值（潜水
	员）正无穷。来表示状态不合法。*/
    memset(f, -0x3f, sizeof f);
    f[0] = 0;
    g[0] = 1;

    for (int i = 0; i < n; i ++ )
    {
        int v, w;
        cin >> v >> w;
        for (int j = m; j >= v; j -- )
        {
            int maxv = max(f[j], f[j - v] + w);
            int s = 0;
            if (f[j] == maxv) s = g[j];
            if (f[j - v] + w == maxv) s = (s + g[j - v]) % mod;
            f[j] = maxv, g[j] = s;
        }
    }
	//寻找最优解 
    int res = 0;
    for (int i = 1; i <= m; i ++ )
        if (f[i] > f[res])
            res = i;
	//寻找最优解的个数
    int sum = 0;
    for (int i = 0; i <= m; i ++ )
        if (f[i] == f[res])
            sum = (sum + g[i]) % mod;

    cout << sum << endl;

    return 0;
}

6、能量石(01 贪心+dp)

#include <iostream>
#include<cstring>
#include <algorithm>

using namespace std;

//n是能量石个数 m是所消耗的总时间
const int N = 110, M = 10010;

int n;
struct Stone
{
    int s, e, l;
}stones[N];

bool cmp(Stone a, Stone b)
{
    return a.s * b.l < b.s * a.l;
}

int f[M];

int main()
{
    int T;
    cin >> T;
    for (int C = 1; C <= T; C ++ )
    {
        cin >> n;
        //总共的时间
        int m = 0;
        for (int i = 1; i <= n; i ++ )
        {
            int s, e, l;
            cin >> s >> e >> l;
            stones[i] = {s, e, l};
            m += s;
        }

        sort(stones + 1, stones + 1 + n, cmp);
        //消耗的时间恰好是j  见潜水员 最大值初始成负无穷
        memset(f,-0x3f,sizeof f);
        f[0]=0;

        for (int i = 1; i <= n; i ++ )
        {
            int s = stones[i].s, e = stones[i].e, l = stones[i].l;
            for (int j = m; j >=s; j -- )
                f[j] = max(f[j], f[j - s] + max(0, e - l * (j - s)));
        }
        //枚举所有的时间
        int res = 0;
        for (int i = 0; i <= m; i ++ ) res = max(res, f[i]);

        printf("Case #%d: %d\n", C, res);
    }

    return 0;
}