这周末要准备算法比赛,现在还太水了,今天才学了01背包,颇有感触,千言万语涌上心头,只好化作行行代码,与各位分享一下。
我们都知道01背包是道很经典的动态规划题。
描述:
01背包:有N件物品和一个重量为M的背包。(每种物品均只有一件)第i件物品的重量是w[i],价值是p[i]。求解将哪些物品装入背包可使价值总和最大。
呃呃呃,我还是直接假设情景再贴代码吧。
情景:
现在,假设我的背包能承受的重量m=10,我现在手中有4样物品,令n=4,物品的重量和价值分别为(5,4),(5,6),(3,3),(6,9)。输出可使价值总和最大的物品。
效果:
啦啦啦,贴代码
public class main {
public static void main(String [] args){
int m=10,n=4;
int w[]={5,5,3,6};
int q[]={4,6,3,9};
main a=new main();
int c[][]=a.pack(m, n, w, q);
a.print(c, m, n, w);
}
public int[][] pack(int m,int n,int w[],int q[]){
int c[][]=new int [n+1][m+1];
for(int i=0;i<n+1;i++){
c[i][0]=0;
}
for(int j=0;j<m+1;j++){
c[0][j]=0;
}
for(int i=1;i<n+1;i++)
{
for(int j=1;j<m+1;j++)
{
if(w[i-1]<=j)
{
if(c[i-1][j]<c[i-1][j-w[i-1]]+q[i-1]){
c[i][j]=c[i-1][j-w[i-1]]+q[i-1];
}
else {
c[i][j]=c[i-1][j];
}
}
else{
c[i][j]=c[i-1][j];
}
}
}
return c;
}
public void print(int c[][],int m,int n,int w[]){
int x[]=new int[n];
for(int i=n;i>0;i--){
if(c[i][m]>c[i-1][m]){
x[i-1]=1;
m-=w[i-1];
}
}
for(int i=0;i<n;i++){
if(x[i]==1){
System.out.print(w[i]+" ");
}
}
}
}