小记: 这题一看 就是dp,但是出题的时候 一同学说都是搜索题,刚开始我用一维dp,发现不行,因为有两个约束条件,我就将dp删了,用dfs做,就情况来说,想的头痛,然后认真分析,这是一个典型的0/1背包,只不过要用个二维数组,约束两个条件,于是一个三重循环,直接AC了。
思路:dp,三重循环,第一重枚举石头,第二重枚举重量,第三重枚举石头个数,因为每种石头只有一个,所以重量和石头个数都是从大到小枚举。背包问题,详情请看背包九讲,我的blog里也有一篇讲解的,点击这里
状态转移方程:dp[w][k] = max(dp[w][k],dp[w - f[i].w][k-1] + f[i].value)
dp数组初始化 为0
代码:
#include <iostream> #include <cstdio> #include <cstring> #include <queue> using namespace std; const int MAX_ = 1001; struct point{ int w,v; } f[MAX_]; int dp[MAX_][MAX_]; int weight, n , k; int main() { int T; cin>>T; while(T--) { cin>>n>>k; for(int i = 1; i <= n; ++i){ cin>>f[i].v>>f[i].w; } cin>>weight; memset(dp,0,sizeof(dp)); for(int i = 1; i <= n; ++i){ for(int j = weight; j > -1; --j){ for(int r = k ; r > -1; --r){ if(j >= f[i].w && r > 0){ dp[j][r] = max(dp[j][r],dp[j - f[i].w][r - 1] + f[i].v); } } } } cout<<dp[weight][k]<<endl; } return 0; }