寝癖頭の解法

小学生の目線から、勉強中の覚え書きを投稿、更新していきます。

アルゴ式(beta版): C++による「【特集】典型的な動的計画法のパターンを整理 ~ ナップサック DP 編 ~」ナップサック問題の解答例

アルゴ式(beta版)の「【特集】典型的な動的計画法のパターンを整理 ~ ナップサック DP 編 ~」ナップサック問題からの出典です。
algo-method.com
アルゴ式とは...
>・プログラミングや情報科学をコツコツ学べる「教科書」
>・学んだ内容をゲーム感覚で大量に実践できる「練習問題」
>の2つで構成される、Web上で完結した学習コンテンツです。

C++による「【特集】典型的な動的計画法のパターンを整理 ~ ナップサック DP 編 ~」ナップサック問題の解答例

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

問題 1:最大和問題

algo-method.com

/*
C++による「【特集】典型的な動的計画法のパターンを整理 ~ ナップサック DP 編 ~」ナップサック問題の解答例
問題 1:最大和問題
https://algo-method.com/tasks/307
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
int main(void){
    int n;
    cin>>n;
    vector<int> a(n);
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    vector<int> dp(n+1);
    dp[0]=0;
    for(int i=0;i<n;i++){
        dp[i+1]=max(dp[i],dp[i]+a[i]);
    }
    cout<<dp[n]<<endl;
    return 0;
}
・問題 2:ナップサック問題

algo-method.com

/*
C++による「【特集】典型的な動的計画法のパターンを整理 ~ ナップサック DP 編 ~」ナップサック問題の解答例
問題 2:ナップサック問題
https://algo-method.com/tasks/308
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
int main(void){
    int N,W;
    cin>>N>>W;
    vector<int> w(N),v(N);
    for(int i=0;i<N;i++){
        cin>>w[i]>>v[i];
    }
    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>=w[i]){
                dp[i+1][j]=max(dp[i][j-w[i]]+v[i],dp[i][j]);
            }else{
                dp[i+1][j]=dp[i][j];
            }
        }
    }
    cout<<dp[N][W]<<endl;
    return 0;
}

設問の出典は、情報科学をコツコツ積み立てて学習できるサービス「アルゴ式(beta版)」です。
algo-method.com