#846736 (C++17) No.409 ダイエット

提出ソース

結果

問題	No.409 ダイエット
ユーザー	cutmdo
提出日時	2023-03-08 04:35:51
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	124 ms / 2,000 ms
コード長	6,034 bytes
コンパイル時間	3,913 ms
コンパイル使用メモリ	278,380 KB
実行使用メモリ	42,440 KB
最終ジャッジ日時	2024-09-18 02:25:22
合計ジャッジ時間	8,679 ms
ジャッジサーバーID （参考情報）	judge1 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	2 ms 6,816 KB
testcase_02	AC	2 ms 6,940 KB
testcase_03	AC	2 ms 6,940 KB
testcase_04	AC	1 ms 6,944 KB
testcase_05	AC	2 ms 6,940 KB
testcase_06	AC	1 ms 6,940 KB
testcase_07	AC	2 ms 6,940 KB
testcase_08	AC	2 ms 6,940 KB
testcase_09	AC	1 ms 6,940 KB
testcase_10	AC	2 ms 6,944 KB
testcase_11	AC	1 ms 6,940 KB
testcase_12	AC	1 ms 6,944 KB
testcase_13	AC	2 ms 6,944 KB
testcase_14	AC	2 ms 6,944 KB
testcase_15	AC	2 ms 6,940 KB
testcase_16	AC	2 ms 6,940 KB
testcase_17	AC	2 ms 6,944 KB
testcase_18	AC	1 ms 6,940 KB
testcase_19	AC	2 ms 6,940 KB
testcase_20	AC	2 ms 6,944 KB
testcase_21	AC	1 ms 6,940 KB
testcase_22	AC	2 ms 6,940 KB
testcase_23	AC	1 ms 6,940 KB
testcase_24	AC	1 ms 6,940 KB
testcase_25	AC	1 ms 6,940 KB
testcase_26	AC	2 ms 6,940 KB
testcase_27	AC	2 ms 6,940 KB
testcase_28	AC	1 ms 6,940 KB
testcase_29	AC	1 ms 6,940 KB
testcase_30	AC	2 ms 6,940 KB
testcase_31	AC	2 ms 6,944 KB
testcase_32	AC	2 ms 6,940 KB
testcase_33	AC	1 ms 6,944 KB
testcase_34	AC	2 ms 6,944 KB
testcase_35	AC	3 ms 6,940 KB
testcase_36	AC	3 ms 6,940 KB
testcase_37	AC	3 ms 6,940 KB
testcase_38	AC	3 ms 6,944 KB
testcase_39	AC	3 ms 6,944 KB
testcase_40	AC	2 ms 6,940 KB
testcase_41	AC	3 ms 6,940 KB
testcase_42	AC	2 ms 6,940 KB
testcase_43	AC	3 ms 6,940 KB
testcase_44	AC	3 ms 6,940 KB
testcase_45	AC	3 ms 6,944 KB
testcase_46	AC	3 ms 6,944 KB
testcase_47	AC	3 ms 6,944 KB
testcase_48	AC	2 ms 6,944 KB
testcase_49	AC	3 ms 6,940 KB
testcase_50	AC	2 ms 6,944 KB
testcase_51	AC	3 ms 6,940 KB
testcase_52	AC	3 ms 6,940 KB
testcase_53	AC	3 ms 6,940 KB
testcase_54	AC	2 ms 6,940 KB
testcase_55	AC	57 ms 22,388 KB
testcase_56	AC	72 ms 41,876 KB
testcase_57	AC	116 ms 42,440 KB
testcase_58	AC	48 ms 16,420 KB
testcase_59	AC	73 ms 25,880 KB
testcase_60	AC	43 ms 15,660 KB
testcase_61	AC	99 ms 30,184 KB
testcase_62	AC	110 ms 41,464 KB
testcase_63	AC	99 ms 30,012 KB
testcase_64	AC	54 ms 22,292 KB
testcase_65	AC	115 ms 40,788 KB
testcase_66	AC	120 ms 42,176 KB
testcase_67	AC	85 ms 27,944 KB
testcase_68	AC	72 ms 25,776 KB
testcase_69	AC	97 ms 29,780 KB
testcase_70	AC	101 ms 29,040 KB
testcase_71	AC	54 ms 16,872 KB
testcase_72	AC	124 ms 42,100 KB
testcase_73	AC	110 ms 30,584 KB
testcase_74	AC	70 ms 23,848 KB
testcase_75	AC	115 ms 40,888 KB
testcase_76	AC	82 ms 26,440 KB
testcase_77	AC	52 ms 16,468 KB
testcase_78	AC	62 ms 22,596 KB
testcase_79	AC	48 ms 16,056 KB
testcase_80	AC	117 ms 41,340 KB
testcase_81	AC	107 ms 30,292 KB
testcase_82	AC	81 ms 26,620 KB
testcase_83	AC	87 ms 27,588 KB
testcase_84	AC	71 ms 23,992 KB
testcase_85	AC	9 ms 6,940 KB
testcase_86	AC	69 ms 24,032 KB
testcase_87	AC	99 ms 28,940 KB
testcase_88	AC	41 ms 14,764 KB
testcase_89	AC	86 ms 26,760 KB
testcase_90	AC	39 ms 14,480 KB
testcase_91	AC	89 ms 27,708 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,avx512f")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <iostream>
#include <iomanip>
#include <string>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#include <list>
#include <stack>
#include <queue>
#include <bitset>
#include <numeric>
#include <cassert>
#include <memory>
#include <random>
#include <functional>
#include <complex>
#include <immintrin.h>
#include <stdexcept>
#ifdef DEBUG
#include "./CompetitiveProgrammingCpp/Utils/debug.hpp"
#include "./CompetitiveProgrammingCpp/Utils/Timer.hpp"
#include "./CompetitiveProgrammingCpp/Utils/sample.hpp"
#else
#define dump(...)
template<class T>constexpr inline auto d_val(T a, T b) { return a; }
#endif

/* macro */
#define FOR(i, b, e) for(ll i = (ll)(b); i < (ll)(e); ++i)
#define RFOR(i, b, e) for(ll i = (ll)((e)-1); i >= (ll)(b); --i)
#define REP(i, n) FOR(i, 0, (n))
#define RREP(i, n) RFOR(i, 0, (n))
#define REPC(x,c) for(const auto& x:(c))
#define REPI2(it,b,e) for(auto it = (b); it != (e); ++it)
#define REPI(it,c) REPI2(it, (c).begin(), (c).end())
#define RREPI(it,c) REPI2(it, (c).rbegin(), (c).rend())
#define REPI_ERACE2(it, b, e) for(auto it = (b); it != (e);)
#define REPI_ERACE(it, c) REPI_ERACE2(it, (c).begin(), (c).end())
#define ALL(x) (x).begin(),(x).end()
#define cauto const auto&
/* macro func */
template<class T>
inline auto sort(T& t) { std::sort(ALL(t)); }
template<class T>
inline auto rsort(T& t) { std::sort((t).rbegin(), (t).rend()); }
template<class T>
inline auto unique(T& t) { (t).erase(unique((t).begin(), (t).end()), (t).end()); }
template<class T, class S>
inline auto chmax(T& t, const S& s) { if(s > t) { t = s; return true; } return false; }
template<class T, class S>
inline auto chmin(T& t, const S& s) { if(s < t) { t = s; return true; } return false; }
inline auto BR() { std::cout << "\n"; }

/* type define */
using ll = long long;
using VS = std::vector<std::string>;
using VL = std::vector<long long>;
using VVL = std::vector<VL>;
using VVVL = std::vector<VVL>;
using VVVVL = std::vector<VVVL>;
using VVVVVL = std::vector<VVVVL>;
using VD = std::vector<double>;
template<class T>
using V = std::vector<T>;
template<class T = ll, class U = T>
using P = std::pair<T, U>;
using PAIR = P<ll>;

/* using std */
using std::cout;
constexpr char endl = '\n';
using std::cin;
using std::pair;
using std::string;
using std::stack;
using std::queue;
using std::deque;
using std::vector;
using std::list;
using std::map;
using std::unordered_map;
using std::multimap;
using std::unordered_multimap;
using std::set;
using std::unordered_set;
using std::unordered_multiset;
using std::multiset;
using std::bitset;
using std::priority_queue;

/* Initial processing  */
struct Preprocessing { Preprocessing() { std::cin.tie(0); std::ios::sync_with_stdio(0); }; }_Preprocessing;

/* Remove the source of the bug */
auto pow(signed, signed) { assert(false); }

/* define hash */
namespace std {
template <>	class hash<std::pair<ll, ll>> { public:	size_t operator()(const std::pair<ll, ll>& x) const { return hash<ll>()(1000000000 * x.first + x.second); } };
}

/* input */
template<class T> std::istream& operator >> (std::istream& is, vector<T>& vec) { for(T& x : vec) is >> x; return is; }

/* constant value */
// constexpr ll MOD = 1000000007;
constexpr ll MOD = 998244353;

//=============================================================================================


class LiChaoTree {
    using T = long long;
    using Line = std::pair<T, T>;
    constexpr static T INF = std::numeric_limits< T >::max() / 2;

    int m_size;
    std::vector<T> m_x;
    std::vector<Line> m_node;
    std::unordered_map<T, T> m_xtoi;

    static inline int calcSize(int n) { int size = 1; while(size < n) { size <<= 1; }return size; }
    auto f(const Line& line, const T& x)const { return line.first * x + line.second; }

    auto add(const Line& line_, int k, int l, int r) {
        auto line = line_;

        auto m = (l + r) / 2;
        if(f(line, m_x[m]) < f(m_node[k], m_x[m])) { std::swap(line, m_node[k]); }
        if(l + 1 == r) { return; }
        if(line.first > m_node[k].first) {
            add(line, (k << 1) + 1, l, m);
        } else if(line.first < m_node[k].first) {
            add(line, (k << 1) + 2, m, r);
        }
    }

public:
    LiChaoTree(const std::vector<T>& x_) :m_size(calcSize(x_.size())) {
        auto x = x_;
        std::sort(x.begin(), x.end());
        x.erase(std::unique(x.begin(), x.end()), x.end());
        m_x = decltype(m_x)(m_size);
        for(size_t i = 0; i < x.size(); ++i) {
            m_x[i] = x[i];
            m_xtoi.emplace(x[i], i);
        }
        for(size_t i = x.size(); i < m_size; ++i) { m_x[i] = m_x[i - 1] + 1; }
        m_node = decltype(m_node)(m_size << 1, {0,INF});
    }

    auto add(const Line& line) { add(line, 0, 0, m_size); }
    auto add(const T& a, const T& b) { add({a,b}); }

    auto query(const T& x) {
        auto k = m_xtoi[x] + m_size;
        auto ret = INF;
        while(k > 0) {
            ret = std::min(ret, f(m_node[k - 1], x));
            k >>= 1;
        }
        return ret;
    }
};

signed main() {
    ll n, a, b, w;
    cin >> n >> a >> b >> w;
    VL d(n);
    cin >> d;

    VL xs(n);
    std::iota(ALL(xs), 1);
    LiChaoTree lct(xs);

    constexpr ll mx = 1e18;
    VL dp(n + 1, mx);
    auto update = [&](ll i, ll x) {
        chmin(dp[i], x);
        auto pa = -b * i;
        auto pb = a * i + i * (i + 1) / 2 * b + dp[i];
        lct.add(pa, pb);
    };

    update(0, w);
    FOR(i, 1, n + 1) {
        auto ad = d[i - 1] - a * i + a + i * (i - 1) / 2 * b;
        auto min = lct.query(i);
        update(i, ad + min);
    }

    ll ans = dp[n];
    REP(i, n) {
        ll k = n - i;
        chmin(ans, dp[i] + -a * k + k * (k + 1) / 2 * b);
    }

    cout << ans << endl;
}

yukicoder

結果

テストケース

ソースコード