动态规划自记录

Lzy0207

已于 2024-04-24 20:09:49 修改

阅读量627

点赞数 17

文章标签：动态规划算法

于 2024-04-22 19:56:45 首次发布

本文链接：https://blog.csdn.net/liu285783/article/details/138080013

版权

本文介绍了C++中针对不同类型的背包问题（0-1背包、完全背包、多重背包、朴素版和二进制优化）的未优化和优化版本代码，展示了如何通过算法优化减少计算复杂度。

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

1.背包问题：

①0-1背包问题：每件物品最多只用一次

未优化版：

#include <iostream>
#include <algorithm>
using namespace std;

const int N=1010;

int n,m;
int v[N],w[N];
int f[N][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=0;j<=m;j++)
        {
            f[i][j] = f[i-1][j];
            if(v[i]<=j) f[i][j] = max(f[i][j],f[i-1][j-v[i]]+w[i]);
        }
    cout<<f[n][m]<<endl;
    return 0;
}

一维优化版：

#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;
}

②完全背包问题：每件物品有无限个

未优化版：

#include <iostream>
#include <algorithm>
using namespace std;
const int N=1010;
int n,m;
int v[N],w[N];
int f[N][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=0;j<=m;j++)
            for(int k=0;k*v[i]<=j;k++)
                f[i][j] = max(f[i][j],f[i-1][j-k*v[i]]+w[i]*k);
    cout<<f[n][m];
    return 0;
}

优化版：

#include <iostream>
#include <algorithm>
using namespace std;
const int N=1010;
int n,m;
int v[N],w[N];
int f[N][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=0;j<=m;j++)
        {
            f[i][j] = f[i-1][j];
            if(v[i]<=j) f[i][j] = max(f[i][j],f[i][j-v[i]]+w[i]);
        }
    cout<<f[n][m];
    return 0;
}

一维优化版：

#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];
    return 0;
}

③多重背包问题：每个物品有有限个，告诉我们有多少个

朴素版：

#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;
}

二进制优化：

#include <iostream>
#include <algorithm>
using namespace std;
const int N=25000,M=2010;
int n,m;
int v[N],w[N];
int f[N]; 
//思想就是把s[i]个数按二进制分组 每一个个数都能用不同的组合表示出来，然后化成01背包问题
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] = k*a;
            w[cnt] = k*b;
            s -= k;
            k *= 2;
        }
        if(s>0)
        {
            cnt++;
            v[cnt] = s*a;
            w[cnt] = s*b;
        }
    }
    n = cnt;
    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];
    return 0;
}

④分组背包问题：每个组里我们最多选一个物品

#include <iostream>
#include <algorithm>
using namespace std;
const int N=110;

int n,m;
int s[N],v[N][N],w[N][N];
int f[N];
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>s[i];
        for(int j=1;j<=s[i];j++)
            cin>>v[i][j]>>w[i][j];
    }
    for(int i=1;i<=n;i++)
        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];
    
    
    return 0;
}