ARC 042 C - おやつ

問題

C: おやつ - AtCoder Regular Contest 042 | AtCoder

回答

CPP
Submission #468830 - AtCoder Regular Contest 042 | AtCoder

同じアルゴリズムで書いてもPythonだとTLE

いわゆるDPを扱うことができれば溶けるタイプの問題
しかし実は私はDPが苦手なので・・・

４時間かかりました．．．

参考にしたところ
動的計画法（ナップサック問題） - アルゴリズム講習会
片鱗懐古のブログ: 01ナップサック問題を動的計画法で解く場合の考え方
Submission #456244 - AtCoder Regular Contest 042 | AtCoder

解説を読んで，実装してテストして．．．
を繰り返し，通っている人のものを参考にし，なんとかACにたどり着きました．．．
解説のスライド
http://www.slideshare.net/chokudai/arc042

解放に関してはスライド読んでください．値段順にソートしてDPすれば良いので．

#include <iostream>
#include <cstdio>
#include <vector>
#include <queue>
 
#include <map>
#include <set>
#include <string>
#include <algorithm>
#include <functional>
using namespace std;
#define FOR(i,a,b) for (int i=(a);i<(b);i++)
#define RFOR(i,a,b) for (int i=(b)-1;i>=(a);i--)
#define REP(i,n) for (int i=0;i<(n);i++)
#define RREP(i,n) for (int i=(n)-1;i>=0;i--)
#define INF 1<<30
#define MP make_pair
#define mp make_pair
#define pb push_back
#define PB push_back
#define DEBUG(x) cout<<#x<<": "<<x<<endl
#define ll long long
#define ull unsigned long long
 
int dp[5005];
 
int main(){
    cin.tie(0);
    ios::sync_with_stdio(false);
    cout.precision(16);
 
    int n, p;
    cin >> n >> p;
 
    vector< pair<int, int> > data(n);
    REP(i, n) {
        cin >> data[i].first >> data[i].second;
    }
 
    sort(data.begin(), data.end());
    reverse(data.begin(), data.end());
 
    int res = 0;
    REP(i, n) {
        int f = data[i].first;
        int s = data[i].second;
        RREP(j, p) {
            if(j+f <= p) {
                dp[j+f] = max(dp[j+f], s+dp[j]);
            }
        }
 
        // 次のを乗っけてみる
        if(i < n-1) res = max(res, dp[p]+data[i+1].second);
        /*  DEBUG
        REP(j, p+1) cout << dp[j] << " ";
        cout << endl;
        DEBUG(res);
        */
    }
 
    cout <<  max(res, dp[p]) << endl;
    return 0;
}

久々に大変な思いをした．圧倒的に成長できていると信じたい．