动态规划解题步骤
1.确定状态变量(函数)
2.确定状态转移方程(递推关系)---关键!!!
3.确定边界条件
例题
题目描述
辰辰是个天资聪颖的孩子,他的梦想是成为世界上最伟大的医师。为此,他想拜附近最有威望的医师为师。医师为了判断他的资质,给他出了一个难题。医师把他带到一个到处都是草药的山洞里对他说:“孩子,这个山洞里有一些不同的草药,采每一株都需要一些时间,每一株也有它自身的价值。我会给你一段时间,在这段时间里,你可以采到一些草药。如果你是一个聪明的孩子,你应该可以让采到的草药的总价值最大。”
如果你是辰辰,你能完成这个任务吗?
输入格式
第一行有 2 个整数 T(1≤T≤1000)和 M(1≤M≤100),用一个空格隔开,T 代表总共能够用来采药的时间,M 代表山洞里的草药的数目。
接下来的 M 行每行包括两个在 1 到 100 之间(包括 1 和 100)的整数,分别表示采摘某株草药的时间和这株草药的价值。
输出格式
输出在规定的时间内可以采到的草药的最大总价值。
在不考虑空间复杂度的情况下,首先我们可用二维数组来做
f[i][j]//表示前i个草药采摘花费时间j下的最大价值
如果j<time[i] 则 f[i][j] = f[i-1][j];
因为你啥也采摘不了
如果j>=time[i] 则 可以采摘,但要考虑到底要不要采摘
f[i][j] = max(f[i-1][j],f[i-1][j-time[i]]+value[i])
#include <bits/stdc++.h>
#define int long long
#define ll long long
#define endl '\n'
using namespace std;
#define ios ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
void solve()
{
int t,m;
cin>>t>>m;
vector<int> time(m+1),value(m+1);
for(int i = 1;i<=m;i++) cin>>time[i]>>value[i];
int f[1005][1005];
for(int i = 1;i<=m;i++)
{
for(int j = 1;j<=t;j++)
{
if(j<time[i]) f[i][j] = f[i-1][j];
else f[i][j] = max(f[i-1][j],f[i-1][j-time[i]]+value[i]);
}
}
cout<<f[m][t]<<endl;
}
signed main(void)
{
ios;
solve();
return 0;
}
这是不考虑空间复杂度的做法,这个的空间复杂度为O((t+1)(m+1));
下面是优化做法
#include <bits/stdc++.h>
#define int long long
#define ll long long
#define endl '\n'
using namespace std;
#define ios ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
void solve()
{
int t,m;
cin>>t>>m;
vector<int> time(m+1),value(m+1),f(t+1);
for(int i = 1;i<=m;i++) cin>>time[i]>>value[i];
for(int i = 1;i<=m;i++)
{
for(int j = t;j>=1;j--)
{
if(time[i]>j) f[j] = f[j];
else f[j] = max(f[j],f[j-time[i]]+value[i]);
}
}
cout<<f[t]<<endl;
}
signed main(void)
{
solve();
return 0;
}
再次压榨性能
#include <bits/stdc++.h>
#define int long long
#define ll long long
#define endl '\n'
using namespace std;
#define ios ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
void solve()
{
int t,m;
cin>>t>>m;
vector<int> time(m+1),value(m+1),f(t+1);
for(int i = 1;i<=m;i++) cin>>time[i]>>value[i];
for(int i = 1;i<=m;i++)
{
for(int j = t;j>=time[i];j--)
{
f[j] = max(f[j],f[j-time[i]]+value[i]);
}
}
cout<<f[t]<<endl;
}
signed main(void)
{
solve();
return 0;
}