最近补题补得有点惨,发现所谓的会写的题以前看过题解的题只要是自己做的话,永远都做不对
所以每次做完一道错题写完一道AC的代码之后都得水一发博客,下次也能记得在哪
分组背包第一题:分组背包
题意:有n门课程m天,矩阵a[i][j]为在第i门课程上花费j天能够获得的价值。现在m天求最大可以收获的价值
本质上是01背包:哪些选择哪些不选
可是细节上有不同:选择的课程中哪些科目花费多少天,这些都是需要制定的
定义二维:dp【i】【j】为在前i门课程中,花费一共j天的最大价值,那么需要由前i-1天的值推出,所以需要知道第i门课程花费了多少天(不得超过最价值的天数,同理可以不要第i门课程,即花费0天)
dp【i】【j】=max{dp【i】【j】,dp【i-1】【j】,dp【i-1】【j-k】+a【i】【k】}
其中呢:
dp【i】【j】表示跟之前算过的所有值做比较取最大
dp【i-1】【j】表示不需要第i门课程
dp【i-1】【j-k】+a【i】【k】在第i门课程上花费了k天,前i-1门课程花费j-k天
因此需要3重循环,而且上下限已定:由于本质是01背包,一定记得j的顺序(逆序,否则变成完全背包)
代码如下:
int n,m;
int dp[maxn][maxn];
int a[maxn][maxn];
int main(){
//input;
int i,j,k;
while(scanf("%d%d",&n,&m)!=EOF){
if (!n&&!m) break;
for(i=1;i<=n;i++)
for(j=1;j<=m;j++)
scanf("%d",&a[i][j]);
memset(dp,0,sizeof(dp));
for(i=1;i<=n;i++)
for(j=m;j>=0;j--)
for(k=1;k<=j;k++){
dp[i][j]=max(dp[i][j],dp[i-1][j-k]+a[i][k]);
dp[i][j]=max(dp[i][j],dp[i-1][j]);
}
printf("%d\n",dp[n][m]);
}
return 0;
}
01背包可以简化到一维,同理分组背包:dp【j】花费j天的最大价值
定义好了之后,注意这个细节:
只有两种情况,选择(dp[j]+a[i][k])或者不选择(dp[j])
因为在当前层循环之前,数组的值就是所有的不取
第i个没循环到的时候把第i个不取的已经算好了
代码如下:
int main()
{
//freopen("in.txt","r",stdin);
int n,m,i,j,k;
while(cin>>n>>m)
{
if(n==0&&m==0)
break;
memset(dp,0,sizeof(dp));
//输入部分
for(i=1;i<=n;i++)
for(j=1;j<=m;j++)
cin>>a[i][j];
for(i=1;i<=n;i++)
for(j=m;j>=0;j--)
for(k=j;k<=m;k++)
dp[j]=max(dp[j],dp[j-k]+a[i][k]);
cout<<dp[m]<<endl;
}
return 0;
}
注意分组背包的实质是01背包,就能够理解为什么会要多一重循环:
01背包是当前物品取或不取,只有两种状态
分组背包是取或者不取,但是取,怎么取,取多少,仍然需要一个循环来枚举