Rabbit尋寶記(2)
阿新 • • 發佈:2019-02-17
Description
Rabbit成功地打開了大門後,沒多久就見到了夢寐以求的寶藏。裡面的寶石種類共有N 種,每一種都有一個體積v 和它的價值val 。(已知第i 種寶石的體積為i ,編號從1 ~N )更讓Rabbit興奮的是,每種寶石的數量還是無窮無盡的。
Rabbit當然想把所有寶石全都帶回家,但是她帶的袋子卻最多隻能裝下總體積為V 的寶石,所以貪心的Rabbit決定要帶走總體積恰好為V 的寶石。
現在她突然想知道自己在拿走寶石數量恰好為N 的情況下總價值最大為多少?
Input
輸入資料第一行是一個正整數T ,表示資料組數。(T<=20 )
每組資料佔兩行。
第一行為兩個整數N,V 。(0<N<=1000,N<=V<=min(N∗N,2∗N) )
接下來一行有N個整數,代表N種寶石的價值vali 。(0<vali<=10000 )
Output
每組測試資料輸出一行,代表在滿足拿走寶石總體積恰為V 和數量恰好為N 的情況下Rabbit能拿走寶石的最大總價值。
Sample Input
1
3 5
6 2 4
Sample Output
16
C++版本一
#include<queue> #include<stack> #include<stdio.h> #include<string.h> #include<stdlib.h> #include<algorithm> using namespace std; int dp[2001][2001]; int dis[2001][2001]; int w[2001]; int v[2001]; int main() { int n,c,t; scanf("%d",&t); while(t--){ scanf("%d%d",&n,&c); int i,j; memset(dp,0,sizeof(dp)); memset(dis,0,sizeof(dis)); for(i=1; i<=n; i++){ scanf("%d",&w[i]); v[i]=i; } for(i=1; i<=n; i++){ for(j=0; j<=c; j++){ for(int k=0; k*v[i]<=j&&k<=n; k++) { if(dis[i][j]+k<=n){ dp[i][j]=max(dp[i][j],dp[i-1][j-k*v[i]]+k*w[i]); dis[i][j]+=k; } } } } printf("%d\n",dp[n][c]); } return 0; }
C++版本二
#include<queue> #include<stack> #include<stdio.h> #include<string.h> #include<stdlib.h> #include<algorithm> using namespace std; int dp[2001]; int dis[2001]; int w[2001]; int v[2001]; int main() { int n,c,t; scanf("%d",&t); while(t--){ scanf("%d%d",&n,&c); int i,j; memset(dp,0,sizeof(dp)); memset(dis,0,sizeof(dis)); for(i=1; i<=n; i++){ scanf("%d",&w[i]); v[i]=i; } for(i=1;i<=n;i++){ for(j=v[i];j<=c;j++){ if(dis[i]+1<=n){ dis[i]+=1; dp[j]=max(dp[j],dp[j-v[i]]+w[i]); } } } printf("%d\n",dp[c]); } return 0; }
以上都是錯解
因為個數的限制我們陣列再加一維
#include<queue>
#include<stack>
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
using namespace std;
int dp[1010][2001];
int dis[2001];
int w[2001];
int v[2001];
int main()
{
int n,c,t;
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&c);
int i,j;
memset(dp,0,sizeof(dp));
memset(dis,0,sizeof(dis));
for(i=1; i<=n; i++){
scanf("%d",&w[i]);
v[i]=i;
}
for(i=1;i<=n;i++){
for(j=v[i];j<=c;j++){
if(dis[i]+1<=n){
dis[i]+=1;
dp[dis[i]][j]=max(dp[dis[i]-1][j],dp[dis[i]-1][j-v[i]]+w[i]);
}
}
}
printf("%d\n",dp[n][c]);
}
return 0;
}