寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

Aizu Online Judge in C++ #DPL_1_B : 0-1 Knapsack Problem

Aizu Online Judge(AOJ)の過去問から、その提出コードの解答例です。

・問題 "0-1 Knapsack Problem"
https://onlinejudge.u-aizu.ac.jp/problems/DPL_1_B
価値が vi 重さが wi であるような N 個の品物と、容量が W のナップザックがあります。
次の条件を満たすように、品物を選んでナップザックに入れます:
・選んだ品物の価値の合計をできるだけ高くする。
・選んだ品物の重さの総和は W を超えない。
価値の合計の最大値を求めてください。

僕が作成、提出したコードは、以下のとおりです。

Aizu Online Judge in C++ #DPL_1_B : 0-1 Knapsack Problem
/*
Aizu Online Judge in C++ #DPL_1_B : 0-1 Knapsack Problem
https://onlinejudge.u-aizu.ac.jp/problems/DPL_1_B
 提出コードの解答例
 https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
int main(void){
    int N,W;
    cin>>N>>W;
    vector<pair<int,int> > vw(N);
    for(int i=0;i<N;i++){
        int v,w;
        cin>>v>>w;
        vw[i].first=v;
        vw[i].second=w;
    }
    vector<vector<int> > dp(N+1,vector<int>(W+1));
    for(int i=0;i<=W;i++){
        dp[0][i]=0;
    }
    for(int i=0;i<N;i++){
        for(int j=0;j<=W;j++){
            if(j>=vw[i].second){
                dp[i+1][j]=max(dp[i][j-vw[i].second]+vw[i].first,dp[i][j]);
            }else{
                dp[i+1][j]=dp[i][j];
            }
        }
    }
    cout<<dp[N][W]<<endl;
    return 0;
}

設問の出典は、プログラミング問題のオンライン採点システム「Aizu Online Judge(AOJ)」です。
http://judge.u-aizu.ac.jp/onlinejudge/