算法基础之背包问题

最新推荐文章于 2023-08-04 14:45:26 发布

素位明哲

最新推荐文章于 2023-08-04 14:45:26 发布

阅读量567

点赞数

分类专栏：算法基础文章标签：算法 c++ 动态规划

本文链接：https://blog.csdn.net/qq_37448336/article/details/121732744

版权

算法基础专栏收录该内容

17 篇文章 2 订阅

订阅专栏

本文介绍了背包问题的几种常见类型，包括01背包问题的动态规划解决方案、优化的一维表示，以及完全背包和多重背包（朴素版与优化版）的算法改进。通过实例展示了如何从二维数组转换为一维滚动数组，提升空间效率。最后，分组背包问题也被详细探讨了其求解过程。

摘要由CSDN通过智能技术生成

1、背包问题

在这里插入图片描述

1.1、01背包问题

在这里插入图片描述

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1010;

int n, m;
int v[N], w[N];
int f[N];

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

    for (int i = 1; i <= n; i ++ ) cin >> v[i] >> w[i];

    for (int i = 1; i <= n; i ++ )
        for (int j = m; j >= v[i]; j -- ) 
            f[j] = max(f[j], f[j - v[i]] + w[i]);

    cout << f[m] << endl;

    return 0;
}

在这里插入图片描述

先写二维表示的动态规划，再优化成一维。

优化

在这里插入图片描述

在这里插入图片描述

滚动数组使空间由二维转化成一维。

1.2、完全背包问题

在这里插入图片描述

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1010;

int n, m;
int v[N], w[N];
int f[N];

int main()
{
    cin >> n >> m;
    for (int i = 1; i <= n; i ++ ) cin >> v[i] >> w[i];

    for (int i = 1; i <= n; i ++ )
        for (int j = v[i]; j <= m; j ++ )
            f[j] = max(f[j], f[j - v[i]] + w[i]);

    cout << f[m] << endl;

    return 0;
}

在这里插入图片描述

二维实现朴素算法

优化成一维

在这里插入图片描述

在这里插入图片描述

将二维优化成一维

在这里插入图片描述

1.3、多重背包问题（朴素版）

在这里插入图片描述

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 110;

int n, m;
int v[N], w[N], s[N];
int f[N][N];

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

    for (int i = 1; i <= n; i ++ ) cin >> v[i] >> w[i] >> s[i];

    for (int i = 1; i <= n; i ++ )
        for (int j = 0; j <= m; j ++ )
            for (int k = 0; k <= s[i] && k * v[i] <= j; k ++ )
                f[i][j] = max(f[i][j], f[i - 1][j - v[i] * k] + w[i] * k);

    cout << f[n][m] << endl;
    return 0;
}

在这里插入图片描述

多重背包的朴素版本与完全背包类似

1.4、多重背包问题2（优化版）

在这里插入图片描述

#include <iostream>
#include <algorithm>

using namespace std;
//log 以2为底
//物品总数 1000*log2000=1000*11（12）+10
const int N = 12010, M = 2010;

int n, m;
int v[N], w[N];
int f[M];

int main()
{
    cin >> n >> m;
	//表示所有新的物品以及编号
    int cnt = 0;
    for (int i = 1; i <= n; i ++ )
    {
    //读入当前物品的体积 价值 个数
        int a, b, s;
        cin >> a >> b >> s;
        int k = 1;
        while (k <= s)
        {
            cnt ++ ;
            v[cnt] = a * k;
            w[cnt] = b * k;
            s -= k;
            k *= 2;
        }
        if (s > 0)
        {
            cnt ++ ;
            v[cnt] = a * s;
            w[cnt] = b * s;
        }
    }
	//二进制划分后的，新的物品数量
    n = cnt;
	//01背包问题
    for (int i = 1;	 i <= n; i ++ )
        for (int j = m; j >= v[i]; j -- )
            f[j] = max(f[j], f[j - v[i]] + w[i]);

    cout << f[m] << endl;

    return 0;
}

#include<iostream>
using namespace std;

const int N=2010;
int f[N];
int n,m;
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++)
    {
        int v,w,s;
        scanf("%d%d%d",&v,&w,&s);
        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);
    }
    printf("%d",f[m]);
    return 0;
}

在这里插入图片描述

1.5、分组背包问题

在这里插入图片描述

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 110;

int n, m;
int v[N][N], w[N][N], s[N];
int f[N];

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

    for (int i = 1; i <= n; i ++ )
    {
        cin >> s[i];
        for (int j = 0; j < s[i]; j ++ )
            cin >> v[i][j] >> w[i][j];
    }
	//从前向后，枚举每一组
    for (int i = 1; i <= n; i ++ )
    	//从大到小枚举体积 本题类似多重背包（01背包）
        for (int j = m; j >= 0; j -- )
        	//枚举所有的选择
            for (int k = 0; k < s[i]; k ++ )
                if (v[i][k] <= j)
                    f[j] = max(f[j], f[j - v[i][k]] + w[i][k]);

    cout << f[m] << endl;

    return 0;
}