Cod sursa(job #2695478)

Utilizator MenelausSontea Vladimir Menelaus Data 13 ianuarie 2021 11:26:37
Problema Problema rucsacului Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.3 kb
#include <iostream>
#include <fstream>

std::ifstream in("rucsac.in");
std::ofstream out("rucsac.out");

const int N=5001;
const int W=10001;

/// dp[i][j] este valoarea maxima ce se poate obtine alegand din primele i obiecte cu suma greutatilor <=j
int dp[2][W];
int val[N];
int wt[N];

int main()
{
    int n, w;
    in>>n>>w;
    for(int i=1; i<=n; i++)
    {
        in>>wt[i]>>val[i];
    }

    for(int i=1; i<=n; i++)
    {
        //rezultatele le stocam pe linia 0
        if(i%2==1)
        {
            for(int j=1; j<=w; j++)
            {
                if(wt[i]<=j)
                {
                    dp[0][j]=std::max(dp[1][j], val[i]+ dp[1][j-wt[i]] );
                }
                else
                {
                    dp[0][j]=dp[1][j];
                }
            }
        }
        else
        {
            for(int j=1; j<=w; j++)
            {
                if(wt[i]<=j)
                {
                    dp[1][j]=std::max(dp[0][j], val[i]+ dp[0][j-wt[i]] );
                }
                else
                {
                    dp[1][j]=dp[0][j];
                }
            }
        }
    }


    switch(n%2)
    {
    case 1:
        out<<dp[0][w];
        break;
    case 0:
        out<<dp[1][w];
    }
}