1. 程式人生 > >二維背包 hdu2159

二維背包 hdu2159

sin name scan size 獲得 表示 clas tac code

題目鏈接:http://acm.hdu.edu.cn/showproblem.php?pid=2159

題目裏面有兩個限制條件,忍耐度和殺怪數量,所以可以用一個二維數組dp[i][j]來表示在消耗忍耐度i,並且殺怪數量為j時可以得到的最大經驗,也可以用三維數組,但是因為每次的答案值依賴前一次的值,所以可以省略一維直接用二維數組。

第一次做這種類型沒想到用二維數組,開了兩個一維數組做,現在有點覺得這個維數是不是就是限制條件的數量,我們用每一維的數組下標來表示每一種限制條件,那以後如果遇到三個限制條件,多個限制條件,那麽是不是應該就是開多維數組來進行計算,感覺又更懂了那麽一點點。

我看博客之後的代碼和第一次寫的代碼:

看了博客之後:

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#include<map>
#include<stack>
#include<cmath>
#include<vector>
#include<set>
#include<cstdio>
#include<string>
#include<deque> 
using namespace std;
typedef 
long long LL; #define eps 1e-8 #define INF 0x3f3f3f3f #define maxn 105 /*struct point{ int u,w; }; bool operator <(const point &s1,const point &s2) { if(s1.w!=s2.w) return s1.w>s2.w; else return s1.u>s2.u; }*/ int dp[maxn][maxn];//dp[i][j]表示忍耐度消耗i,殺死怪物數量為j時可以獲得的最大經驗 int
v[maxn],w[maxn]; int n,m,k,t,s; int main() { while(scanf("%d%d%d%d",&n,&m,&k,&s)!=EOF) { for(int i=1;i<=k;i++) scanf("%d%d",&v[i],&w[i]); memset(dp,0,sizeof(dp)); int ans=-1; for(int i=1;i<=m;i++)//忍耐度,第一個限制條件,這三個循環可以互換位置,但是因為題目要我們求 { //最後剩余的最大忍耐值,所以忍耐度放在第一層循環應該更快一點,可以跳出 for(int j=1;j<=s;j++)//殺怪數量,第二個限制條件 { for(int e=1;e<=k;e++)//怪物種類 { if(i>=w[e]) dp[i][j]=max(dp[i][j],dp[i-w[e]][j-1]+v[e]); } if(dp[i][j]>=n) { ans=i; break; } } if(ans!=-1) break; } if(ans!=-1) printf("%d\n",m-ans); else printf("%d\n",ans); } return 0; }

開了兩個一維數組的代碼:

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#include<map>
#include<stack>
#include<cmath>
#include<vector>
#include<set>
#include<cstdio>
#include<string>
#include<deque> 
using namespace std;
typedef long long LL;
#define eps 1e-8
#define INF 0x3f3f3f3f
#define maxn 105
/*struct point{
    int u,w;
};
bool operator <(const point &s1,const point &s2)
{
    if(s1.w!=s2.w)
    return s1.w>s2.w;
    else
    return s1.u>s2.u;
}*/
int n,m,k,s,t;
int v[maxn],w[maxn],dp[maxn],num[maxn]; 
int main()
{
    while(scanf("%d%d%d%d",&n,&m,&k,&s)!=EOF)
    {
        for(int i=1;i<=k;i++)
        scanf("%d%d",&v[i],&w[i]);
        memset(dp,0,sizeof(dp));//dp[i]表示消耗忍耐度i時可以得到的最大經驗值 
        memset(num,0,sizeof(num));//記錄殺怪數量,兩個數組同步 
        for(int i=1;i<=k;i++)
        {
            for(int j=w[i];j<=m;j++)
            {
                if(num[j-w[i]]+1<=s&&(dp[j]<dp[j-w[i]]+v[i]||dp[j]==dp[j-w[i]]+v[i]&&num[j]>num[j-w[i]]+1))
                {//經驗值放在第一位,然後殺怪數量第二位,但是前提是不能超出s 
                    dp[j]=dp[j-w[i]]+v[i];
                    num[j]=num[j-w[i]]+1;
                }
            }
        }
        int ans=-1;
        for(int i=1;i<=m;i++)//找滿足條件並且消耗最小忍耐度的結果 
        {
            if(dp[i]>=n)
            {
                ans=m-i;
                break;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
} 

二維背包 hdu2159