计算机算法--动态规划0-1背包问题

动态规划 0-1 背包问题

示例输入（规定物品数量为10，背包容量为50，输入为20个数，前十个为物品重量，后十个数为物品价值）：

12

3

11

5

6

8

9

4

7

10

6

2

7

3

2

9

8

10

4

5

示例输出（11个数，第1个数为最大价值，后10个数为解向量）：

44

0

1

1

1

0

1

1

1

0

1

 

import java.util.Arrays;
import java.util.Scanner;

public class niuqingshan {
	public static void knapsack(int[] v, int[] w, int c, int[][] m)
    {          
        int n = v.length-1;
        /** v[] w[] c 分别是价值、重量数组和背包容量,         m[i][j]表示只有w[i],w[i+1]...w[n]这些物品时，背包容量为j时的最大价值。*/
       
        int jMax = Math.min(w[n]-1, c);
        for(int j = 0; j <= jMax; j++)  
            m[n][j] = 0;        //当w[n]>j 有 m[n][j]=0
        //m[n][j] 表示只有w[n]物品，背包的容量为j时的最大价值
        for (int l = w[n]; l <= c; l++)
            m[n][l] = v[n];  //当w[n]<=j 有m[n][j]=v[n]
        //递规调用求出m[][]其它值，直到求出m[0][c]
        for(int i = n-1; i >=1; i--)
        {
            jMax = Math.min(w[i]-1,c);            
            for(int k = 0; k <=jMax; k++)
                m[i][k] = m[i+1][k];
                      
            for(int h = w[i]; h <= c; h++)
                m[i][h] = Math.max(m[i+1][h],m[i+1][h-w[i]]+v[i]);
        }
        m[0][c] = m[1][c];
        if(c >= w[0])
            m[0][c] = Math.max(m[0][c],m[1][c-w[0]]+v[0]);
        System.out.println(m[0][c]);
    }
         
    public static void traceback(int[][] m, int[] w, int c, int[] x)
    {// 根据最优值求出最优解
        int n = w.length-1;
        for(int i = 0; i< n;i++)
            if(m[i][c] == m[i+1][c])
                x[i] = 0;
            else{
                x[i] = 1;
                c -= w[i];
            }
        x[n] = (m[n][c]>0)?1:0;
    }
    public static void main(String[] args)
    {
        //测试
        int[] ww = {12,3,11,5,6,8,9,4,7,10};//int w[] = new int[] {12,3,11,5,6,8,9,4,7,10};  

	                           //int v[] = new int[] { 6,2,7,3,2,9,8,10,4,5 };  
        int[] vv = {6,2,7,3,2,9,8,10,4,5 };
        int[][] mm = new int[51][51];
        knapsack(vv,ww,50,mm);
        int[] xx =new int[ww.length];
        traceback(mm,ww,50,xx);
        for(int i = 0;i< xx.length;i++)
            System.out.println(xx[i]);
    }

}