01背包問題:
一個背包總容量為V,現在有N個物品,第i個 物品體積為weight[i],價值為value[i],現在往背包里面裝東西,怎么裝能使背包的內物品價值最大?
我們可以這樣考慮:在物品比較少,背包容量比較小時怎么解決?用一個數組f[i][j]表示,在只有i個物品,容量為j的情況下背包問題的最優解,那么當物品種類變大為i+1時,最優解是什么?第i+1個物品可以選擇放進背包或者不放進背包(這也就是0和1),假設放進背包(前提是放得下),那么f[i+1][j]=f[i][j-weight[i+1]+value[i+1];如果不放進背包,那么f[i+1][j]=f[i][j]。
由此得出狀態轉移方程:
f[i+1][j]=max(f[i][j],f[i][j-weight[i+1]+value[i+1])
給出一段代碼:
int f[10][2000];//全局變量,自動初始化為0 int weight[10]; int value[10]; #define max(x,y) (x)>(y)?(x):(y) int main() { int N,M; cin>>N;//物品個數 cin>>M;//背包容量 for (int i=1;i<=N; i++) { cin>>weight[i]>>value[i]; } for (int i=1; i<=N; i++) for (int j=1; j<=M; j++) { if (weight[i]<=j) { f[i][j]=max(f[i-1][j],f[i-1][j-weight[i]]+value[i]);//狀態轉移方程 } else f[i][j]=f[i-1][j]; } cout<<f[N][M]<<endl; } 上面的代碼用了一個二維數組存儲,下面試著用一維數組存儲int f[2000];//全局變量,自動初始化為0
int weight[10];
int value[10];
#define max(x,y) (x)>(y)?(x):(y)
int main()
{
int N,M;
cin>>N;//物品個數
cin>>M;//背包容量
for (int i=1;i<=N; i++)
{
cin>>weight[i]>>value[i];
}
for (int i=1; i<=N; i++)
for (int j=M; j>=1; j--)
{
if (weight[i]<=j)
{
f[j]=max(f[j],f[j-weight[i]]+value[i]);
}
}
cout<<f[M]<<endl;//輸出最優解
}
新聞熱點
疑難解答