动态规划 01背包 c算法

#include <stdio.h>
#include <stdlib.h>

#define N 5
#define C 10

void knapsack(int w[N], int v[N], int m[][C]);
void traceback(int m[][C], int w[N], int x[N]);

void knapsack(int w[N], int v[N], int m[][C])
{
    int n = N;
    int c = C;

    int i, j;
    for(i = 0; i < c; i++){
        if(i < w[0]){
            m[0][i] = 0;
        }
        else{
            m[0][i] = v[0];
        }
       
    }

    for(i = 1; i< n-1; i++){
        for(j = 0; j< c; j++){
            if(j > w[i]){
                m[i][j] = max(m[i-1][j], v[i] + m[i-1][j-w[i]]);
            }
            else{
                m[i][j] = m[i-1][j];
            }
        }
    }
    m[n-1][c-1] = m[n-2][c-1];
    if(c > w[n-1]){
        m[n-1][c-1] = max(m[n-2][c-1], v[n-1] + m[n-2][c-w[n-1]]);
    }

}


void traceback(int m[][C], int w[N], int x[N])
{
    int j = C;
    int i;
    for(i = N-1; i >= 1; i--){
        if(m[i][j-1] == m[i-1][j-1]){
            x[i] = 0;
        }
        else{
            j = j - w[i];
            x[i] = 1;

        }
    }
    if(j > 0){
        x[0] = 1;
    }
    else{
        x[0] = 0;
    }

}

int main()
{
    int w[N] = {2, 2, 6, 5, 4};
    int v[N] = {6, 3, 5, 4, 6};
    int x[N] = {0};
    int m[N][C] = {{0}};
    int i;
    knapsack(w, v, m);
    traceback(m, w, x);

    for(i=0; i< N; i++)
    {
      printf("%d ", x[i]);
    }
    printf("/n");

    system("pause");
    return 0;
}

你可能感兴趣的:(动态规划 01背包 c算法)