最近补题补得有点惨,发现所谓的会写的题以前看过题解的题只要是自己做的话,永远都做不对
所以每次做完一道错题写完一道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背包是当前物品取或不取,只有两种状态
分组背包是取或者不取,但是取,怎么取,取多少,仍然需要一个循环来枚举