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

提出ソース

結果

問題	No.409 ダイエット
ユーザー	arktan763
提出日時	2020-08-06 09:17:06
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	134 ms / 2,000 ms
コード長	8,309 bytes
コンパイル時間	2,052 ms
コンパイル使用メモリ	209,568 KB
実行使用メモリ	14,260 KB
最終ジャッジ日時	2024-09-19 07:17:08
合計ジャッジ時間	7,485 ms
ジャッジサーバーID （参考情報）	judge3 / judge4

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,376 KB
testcase_02	AC	2 ms 5,376 KB
testcase_03	AC	1 ms 5,376 KB
testcase_04	AC	2 ms 5,376 KB
testcase_05	AC	1 ms 5,376 KB
testcase_06	AC	2 ms 5,376 KB
testcase_07	AC	1 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	2 ms 5,376 KB
testcase_10	AC	2 ms 5,376 KB
testcase_11	AC	1 ms 5,376 KB
testcase_12	AC	2 ms 5,376 KB
testcase_13	AC	1 ms 5,376 KB
testcase_14	AC	2 ms 5,376 KB
testcase_15	AC	1 ms 5,376 KB
testcase_16	AC	2 ms 5,376 KB
testcase_17	AC	2 ms 5,376 KB
testcase_18	AC	2 ms 5,376 KB
testcase_19	AC	1 ms 5,376 KB
testcase_20	AC	1 ms 5,376 KB
testcase_21	AC	2 ms 5,376 KB
testcase_22	AC	2 ms 5,376 KB
testcase_23	AC	2 ms 5,376 KB
testcase_24	AC	1 ms 5,376 KB
testcase_25	AC	2 ms 5,376 KB
testcase_26	AC	2 ms 5,376 KB
testcase_27	AC	2 ms 5,376 KB
testcase_28	AC	2 ms 5,376 KB
testcase_29	AC	1 ms 5,376 KB
testcase_30	AC	2 ms 5,376 KB
testcase_31	AC	2 ms 5,376 KB
testcase_32	AC	2 ms 5,376 KB
testcase_33	AC	1 ms 5,376 KB
testcase_34	AC	2 ms 5,376 KB
testcase_35	AC	3 ms 5,376 KB
testcase_36	AC	3 ms 5,376 KB
testcase_37	AC	3 ms 5,376 KB
testcase_38	AC	3 ms 5,376 KB
testcase_39	AC	3 ms 5,376 KB
testcase_40	AC	3 ms 5,376 KB
testcase_41	AC	3 ms 5,376 KB
testcase_42	AC	3 ms 5,376 KB
testcase_43	AC	3 ms 5,376 KB
testcase_44	AC	3 ms 5,376 KB
testcase_45	AC	3 ms 5,376 KB
testcase_46	AC	3 ms 5,376 KB
testcase_47	AC	4 ms 5,376 KB
testcase_48	AC	3 ms 5,376 KB
testcase_49	AC	3 ms 5,376 KB
testcase_50	AC	2 ms 5,376 KB
testcase_51	AC	3 ms 5,376 KB
testcase_52	AC	3 ms 5,376 KB
testcase_53	AC	3 ms 5,376 KB
testcase_54	AC	3 ms 5,376 KB
testcase_55	AC	51 ms 8,448 KB
testcase_56	AC	55 ms 9,948 KB
testcase_57	AC	101 ms 14,260 KB
testcase_58	AC	42 ms 7,936 KB
testcase_59	AC	61 ms 9,088 KB
testcase_60	AC	39 ms 7,424 KB
testcase_61	AC	91 ms 12,800 KB
testcase_62	AC	96 ms 13,952 KB
testcase_63	AC	87 ms 12,416 KB
testcase_64	AC	47 ms 8,064 KB
testcase_65	AC	90 ms 11,264 KB
testcase_66	AC	100 ms 13,308 KB
testcase_67	AC	76 ms 11,264 KB
testcase_68	AC	61 ms 8,832 KB
testcase_69	AC	89 ms 12,032 KB
testcase_70	AC	113 ms 12,032 KB
testcase_71	AC	62 ms 8,192 KB
testcase_72	AC	134 ms 13,952 KB
testcase_73	AC	122 ms 11,776 KB
testcase_74	AC	74 ms 9,728 KB
testcase_75	AC	120 ms 12,288 KB
testcase_76	AC	86 ms 10,496 KB
testcase_77	AC	56 ms 7,296 KB
testcase_78	AC	66 ms 9,088 KB
testcase_79	AC	53 ms 7,424 KB
testcase_80	AC	125 ms 13,568 KB
testcase_81	AC	120 ms 11,904 KB
testcase_82	AC	89 ms 10,496 KB
testcase_83	AC	99 ms 10,880 KB
testcase_84	AC	77 ms 8,576 KB
testcase_85	AC	10 ms 5,376 KB
testcase_86	AC	75 ms 8,704 KB
testcase_87	AC	109 ms 12,032 KB
testcase_88	AC	42 ms 6,528 KB
testcase_89	AC	89 ms 8,448 KB
testcase_90	AC	41 ms 5,760 KB
testcase_91	AC	96 ms 8,704 KB

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

ソースコード

raw source code

#include <bits/stdc++.h>

using namespace std;

#define int long long
// #define double long double
#define FOR(i, a, b) for(int i = (a); i < (b); ++i)
#define FORR(i, a, b) for(int i = (a); i > (b); --i)
#define REP(i, n) for(int i = 0; i < (n); ++i)
#define REPR(i, n) for(int i = n; i >= 0; i--)
#define FOREACH(x, a) for(auto &(x) : (a))
#define VECCIN(x)                                                              \
    for(auto &youso_ : (x)) cin >> youso_
#define bitcnt(x) __builtin_popcount(x)
#define lbit(x) __builtin_ffsll(x)
#define rbit(x) (64 - __builtin_clzll(x))
#define fi first
#define se second
#define All(a) (a).begin(), (a).end()
#define rAll(a) (a).rbegin(), (a).rend()
#define cinfast() cin.tie(0), ios::sync_with_stdio(false)
#define precise(x) cout << fixed << setprecision(x)
#define PERM(c)                                                                \
    sort(All(c));                                                              \
    for(bool cp = true; cp; cp = next_permutation(All(c)))
#define COMB(n, k)                                                             \
    for(ll bit = (1LL << k) - 1; bit < (1LL << n); bit = next_combination(bit))
#define PERM2(n, k)                                                            \
    COMB(n, k) {                                                               \
        vector<int> sel;                                                       \
        for(int bitindex = 0; bitindex < n; bitindex++)                        \
            if(bit >> bitindex & 1) sel.emplace_back(bitindex);                \
        PERM(sel) { Printv(sel); }                                             \
    }
#define MKORDER(n)                                                             \
    vector<int> od(n);                                                         \
    iota(All(od), 0LL);

template <typename T = long long> inline T IN() {
    T x;
    cin >> x;
    return (x);
}
inline void CIN() {}
template <class Head, class... Tail>
inline void CIN(Head &&head, Tail &&... tail) {
    cin >> head;
    CIN(move(tail)...);
}
template <class Head> inline void COUT(Head &&head) { cout << (head) << "\n"; }
template <class Head, class... Tail>
inline void COUT(Head &&head, Tail &&... tail) {
    cout << (head) << " ";
    COUT(forward<Tail>(tail)...);
}

#define CCIN(...)                                                              \
    char __VA_ARGS__;                                                          \
    CIN(__VA_ARGS__)
#define DCIN(...)                                                              \
    double __VA_ARGS__;                                                        \
    CIN(__VA_ARGS__)
#define LCIN(...)                                                              \
    long long __VA_ARGS__;                                                     \
    CIN(__VA_ARGS__)
#define SCIN(...)                                                              \
    string __VA_ARGS__;                                                        \
    CIN(__VA_ARGS__)
#define Printv(v)                                                              \
    {                                                                          \
        REP(hoge, v.size())                                                    \
        cout << v[hoge] << (hoge == v.size() - 1 ? "" : " ");                  \
        cout << "\n";                                                          \
    }
template <typename T = string> inline void eputs(T s) {
    cout << s << "\n";
    exit(0);
}
template <typename A, size_t N, typename T>
void Fill(A (&array)[N], const T &val) {
    std::fill((T *)array, (T *)(array + N), val);
}

long long next_combination(long long sub) {
    long long x = sub & -sub, y = sub + x;
    return (((sub & ~y) / x) >> 1) | y;
}
template <class T> inline bool chmax(T &a, const T &b) {
    if(a < b) {
        a = b;
        return 1;
    }
    return 0;
}
template <class T> inline bool chmin(T &a, const T &b) {
    if(a > b) {
        a = b;
        return 1;
    }
    return 0;
}

// generic lambdas
template <typename first>
#if defined(__has_cpp_attribute) && __has_cpp_attribute(nodiscard)
[[nodiscard]]
#elif defined(__GNUC__) &&                                                     \
    (__GNUC__ > 3 || __GNUC__ == 3 && __GNUC_MINOR__ >= 4)
__attribute__((warn_unused_result))
#endif // defined(__has_cpp_attribute) && __has_cpp_attribute(nodiscard)
    static inline constexpr decltype(auto)
    fix(first &&f) noexcept {
    return [f = std::forward<first>(f)](auto &&... args) {
        return f(f, std::forward<decltype(args)>(args)...);
    };
}

template <typename T> using PQG = priority_queue<T, vector<T>, greater<T>>;
template <typename T> using PQ = priority_queue<T>;

typedef long long ll;
typedef vector<ll> VL;
typedef vector<VL> VVL;
typedef pair<ll, ll> PL;
typedef vector<PL> VPL;
typedef vector<bool> VB;
typedef vector<double> VD;
typedef vector<string> VS;

const int INF = 1e9;
const int MOD = 1e9 + 7;
// const int MOD = 998244353;
const ll LINF = 1e18;
const ll dw[] = {1, 1, 0, -1, -1, -1, 0, 1};
const ll dh[] = {0, 1, 1, 1, 0, -1, -1, -1};
#define PI 3.141592653589793230
#define EPS 1e-7

// monotonic
template <typename T, bool isMin> struct ConvexHullTrick {
    using P = pair<T, T>;
    deque<P> H;
    bool empty() const { return H.empty(); }
    void clear() { H.clear(); }

    inline int sgn(T x) { return x == 0 ? 0 : (x < 0 ? -1 : 1); }

    using D = long double;
    inline bool check(const P &a, const P &b, const P &c) {
        if(b.second == a.second || c.second == b.second)
            return sgn(b.first - a.first) * sgn(c.second - b.second) >=
                   sgn(c.first - b.first) * sgn(b.second - a.second);

        // return (b.first-a.first)*(c.second-b.second) >=
        // (b.second-a.second)*(c.first-b.first);
        return D(b.first - a.first) * sgn(c.second - b.second) /
                   D(abs(b.second - a.second)) >=
               D(c.first - b.first) * sgn(b.second - a.second) /
                   D(abs(c.second - b.second));
    }

    void addLine(T m, T b) {
        if(!isMin) m *= -1, b *= -1;
        P line(m, b);
        if(empty()) {
            H.emplace_front(line);
            return;
        }
        if(H.front().first <= m) {
            if(H.front().first == m) {
                if(H.front().second <= b) return;
                H.pop_front();
            }
            while(H.size() >= 2 && check(line, H.front(), H[1])) H.pop_front();
            H.emplace_front(line);
        } else {
            assert(m <= H.back().first);
            if(H.back().first == m) {
                if(H.back().second <= b) return;
                H.pop_back();
            }
            while(H.size() >= 2 && check(H[H.size() - 2], H.back(), line))
                H.pop_back();
            H.emplace_back(line);
        }
    }

    inline T getY(const P &a, const T &x) { return a.first * x + a.second; }

    T query(T x) {
        assert(!empty());
        int l = -1, r = H.size() - 1;
        while(l + 1 < r) {
            int m = (l + r) >> 1;
            if(getY(H[m], x) >= getY(H[m + 1], x))
                l = m;
            else
                r = m;
        }
        if(isMin) return getY(H[r], x);
        return -getY(H[r], x);
    }

    T queryMonotoneInc(T x) {
        assert(!empty());
        while(H.size() >= 2 && getY(H.front(), x) >= getY(H[1], x))
            H.pop_front();
        if(isMin) return getY(H.front(), x);
        return -getY(H.front(), x);
    }

    T queryMonotoneDec(T x) {
        assert(!empty());
        while(H.size() >= 2 && getY(H.back(), x) >= getY(H[H.size() - 2], x))
            H.pop_back();
        if(isMin) return getY(H.back(), x);
        return -getY(H.back(), x);
    }
};

ll dp[300003];
ll ans[300003];

signed main() {
    LCIN(N, A, B, W);
    VL D(N);
    VECCIN(D);
    ConvexHullTrick<ll, true> cht;
    cht.addLine(0, 2 * W);
    FOR(i, 1, N + 1) {
        dp[i] =
            cht.query(i - 1) + i * (i - 1) * B - 2 * A * (i - 1) + 2 * D[i - 1];
        ans[i] = min(dp[i], cht.query(i) + (i + 1) * i * B - 2 * A * i);
        cht.addLine(-2 * i * B, (i * i - i) * B + 2 * A * i + dp[i]);
    }
    cout << ans[N] / 2 << "\n";
}

yukicoder

結果

テストケース

ソースコード