Yukicoder No.115 遠足のおやつ

,

https://yukicoder.me/problems/no/115

貪欲ができる問題でDPを書くの、競プロに慣れたなあという感じがする。 最適性の原理は強力。

solution

素直にDP。$\mathrm{dp}: (N+1) \times (D+1) \times (K+1) \to \mathcal{P}(N+1 \setminus { 0 })$。$O(NDK)$。

貪欲で$O(K)$でもできるらしい。

implementation

#include <iostream>
#include <vector>
#define repeat(i,n) for (int i = 0; (i) < int(n); ++(i))
#define repeat_from(i,m,n) for (int i = (m); (i) < int(n); ++(i))
#define repeat_reverse(i,n) for (int i = (n)-1; (i) >= 0; --(i))
using namespace std;
template <class T> void setmin(T & a, T const & b) { if (b < a) a = b; }
template <typename X, typename T> auto vectors(X x, T a) { return vector<T>(x, a); }
template <typename X, typename Y, typename Z, typename... Zs> auto vectors(X x, Y y, Z z, Zs... zs) { auto cont = vectors(y, z, zs...); return vector<decltype(cont)>(x, cont); }
const vector<int> inf { int(1e9+7) };
int main() {
    int n, d, k; cin >> n >> d >> k;
    auto dp = vectors(d+1, k+1, inf);
    dp[0][0] = {};
    repeat_from (a,1,n+1) {
        repeat_reverse (i,d) if (i+a < d+1) {
            repeat (j,k) {
                vector<int> it = dp[i][j];
                it.push_back(a);
                setmin(dp[i+a][j+1], it);
            }
        }
    }
    if (dp[d][k] == inf) {
        cout << -1 << endl;
    } else {
        repeat (i, dp[d][k].size()) {
            if (i) cout << ' ';
            cout << dp[d][k][i];
        }
        cout << endl;
    }
    return 0;
}