【SSL】1045采药(2005年分区联赛普级组之三)(01背包)(DP)

【SSL】1045采药(2005年分区联赛普级组之三)(01背包)(DP)

Time Limit:1000MS

Memory Limit:65536K

Description

辰辰是个天资聪颖的孩子，他的梦想是成为世界上最伟大的医师。为此，他想拜附近最有威望的医师为师。医师为了判断他的资质，给他出了一个难题。医师把他带到一个到处都是草药的山洞里对他说：“孩子，这个山洞里有一些不同的草药，采每一株都需要一些时间，每一株也有它自身的价值。我会给你一段时间，在这段时间里，你可以采到一些草药。如果你是一个聪明的孩子，你应该可以让采到的草药的总价值最大。”
　　如果你是辰辰，你能完成这个任务吗？

Input

输入的第一行有两个整数T（1 <= T <= 1000）和M（1 <= M <= 100），用一个空格隔开，T代表总共能够用来采药的时间，M代表山洞里的草药的数目。接下来的M行每行包括两个在1到100之间（包括1和100）的整数，分别表示采摘某株草药的时间和这株草药的价值。

Output

输出包括一行，这一行只包含一个整数，表示在规定的时间内，可以采到的草药的最大总价值。

Sample Input

70 3
71 100
69 1
1 2

Sample Output

3

Hint

对于30%的数据，M <= 10；
对于全部的数据，M <= 100。

思路

这道题可以用深搜枚举，时间复杂度O(2ⁿ)，当n>26时会超时。我们可以列一个表格，行表示处理第几种草药，列表示总时间，里面的值表示当前情况下的最大价值。
当输入数据为：
10 4
5 8
3 6
4 7
2 5
列出的表格为：

每种草药只有2种情况，要么选，要么不选。
不选时，当前值=上一行当前列的值。
选时，当前值=上一行(当前列-采当前草药所需时间)+采当前草药所得价值
我们可以看出状态转移方程为：f[i][j]=max(f[i-1][j],f[i-1][j-t[i]]+w[i])，f[0][j]=0,f[i][0]=0,1<=i<=m,1<=j<=t，t为时间，w为价值，答案为f[m][t]。

如果要输出采哪些草药，可以从后往前，比较当前值是否等于上一行当前列的值，相等则没选。

还可以压缩空间
可以用滚动数组
也可以用迭代法

代码

二维数组：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cstdlib> 
using namespace std;
int alltime,m,t[200],p[200],f[200][1010];
void input()//输入
{
 int i;
 scanf("%d%d",&alltime,&m);
 for(i=1;i<=m;i++)
  scanf("%d%d",&t[i],&p[i]);
 return;
}
void DP()
{
 int i,j;
 for(i=1;i<=m;i++)//从前往后推
 {
  for(j=1;j<=alltime;j++)//计算当前情况下的最大价值
  {
   if (j<t[i])
    f[i][j]=f[i-1][j];
   else
    f[i][j]=max(f[i-1][j],f[i-1][j-t[i]]+p[i]);//状态转移方程
  }
 }
 printf("%d",f[m][alltime]);
 return;
}
int main()
{
 input();
 DP();
 return 0;
} 

输出采哪些草药：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cstdlib> 
using namespace std;
int alltime,m,t[200],p[200],f[200][1010],b[200];
void input()//输入
{
 int i;
 scanf("%d%d",&alltime,&m);
 for(i=1;i<=m;i++)
  scanf("%d%d",&t[i],&p[i]);
 return;
}
void DP()
{
 int i,j;
 for(i=1;i<=m;i++)//从前往后推
 {
  for(j=t[i];j<=alltime;j++)//计算当前情况下的最大价值
  {
   f[i][j]=max(f[i-1][j],f[i-1][j-t[i]]+p[i]);//状态转移方程
  }
 }
 printf("%d\n",f[m][alltime]);
 for(i=m;i>=1;i--)
 {  
  if (f[i][alltime]!=f[i-1][alltime])//判断价值是否变化
  {
   alltime-=t[i];//减去所需时间
   b[i]=true;//标记
  }
 }
 for(i=1;i<=m;i++)//输出采哪些草药
  if (b[i])
   printf("%d ",i);
 return;
}
int main()
{
 input();
 DP();
 return 0;
} 

滚动数组：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cstdlib> 
using namespace std;
int alltime,m,t[200],p[200],f[2][1010];
void input()//输入
{
 int i;
 scanf("%d%d",&alltime,&m);
 for(i=1;i<=m;i++)
  scanf("%d%d",&t[i],&p[i]);
 return;
}
void DP()
{
 int i,j;
 for(i=1;i<=m;i++)//从前往后推
 {
  for(j=1;j<=alltime;j++)//计算当前情况下的最大价值
  {
   if (j<t[i])
    f[i%2][j]=f[!(i%2)][j];//通过i%2滚动数组
   else
    f[i%2][j]=max(f[!(i%2)][j],f[!(i%2)][j-t[i]]+p[i]);//状态转移方程
  }
 }
 printf("%d",f[!(i%2)][alltime]);
 return;
}
int main()
{
 input();
 DP();
 return 0;
} 

一维数组(迭代法)：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cstdlib> 
using namespace std;
int alltime,m,t[200],p[200],f[1010];
void input()//输入
{
 int i;
 scanf("%d%d",&alltime,&m);
 for(i=1;i<=m;i++)
  scanf("%d%d",&t[i],&p[i]);
 return;
}
void DP()
{
 int i,j;
 for(i=1;i<=m;i++)//从前往后推
 {
  for(j=alltime;j>=1;j--)//计算当前情况的最大价值
  {
   if (j>=t[i])
    f[j]=max(f[j],f[j-t[i]]+p[i]);//状态转移方程
  }
 }
 printf("%d",f[alltime]);
 return;
}
int main()
{
 input();
 DP();
 return 0;
}