#231323 (C++14) No.409 ダイエット

提出ソース

結果

問題	No.409 ダイエット
ユーザー	Yang33
提出日時	2018-01-22 15:56:12
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	465 ms / 2,000 ms
コード長	4,340 bytes
コンパイル時間	1,856 ms
コンパイル使用メモリ	184,512 KB
実行使用メモリ	43,504 KB
最終ジャッジ日時	2024-06-26 10:27:23
合計ジャッジ時間	14,065 ms
ジャッジサーバーID （参考情報）	judge1 / judge4

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	1 ms 6,940 KB
testcase_02	AC	2 ms 6,940 KB
testcase_03	AC	2 ms 6,944 KB
testcase_04	AC	2 ms 6,944 KB
testcase_05	AC	2 ms 6,940 KB
testcase_06	AC	2 ms 6,940 KB
testcase_07	AC	2 ms 6,944 KB
testcase_08	AC	2 ms 6,944 KB
testcase_09	AC	2 ms 6,944 KB
testcase_10	AC	2 ms 6,944 KB
testcase_11	AC	2 ms 6,944 KB
testcase_12	AC	2 ms 6,940 KB
testcase_13	AC	2 ms 6,944 KB
testcase_14	AC	2 ms 6,940 KB
testcase_15	AC	2 ms 6,940 KB
testcase_16	AC	2 ms 6,944 KB
testcase_17	AC	2 ms 6,940 KB
testcase_18	AC	2 ms 6,940 KB
testcase_19	AC	1 ms 6,944 KB
testcase_20	AC	2 ms 6,944 KB
testcase_21	AC	2 ms 6,940 KB
testcase_22	AC	2 ms 6,940 KB
testcase_23	AC	1 ms 6,940 KB
testcase_24	AC	2 ms 6,940 KB
testcase_25	AC	2 ms 6,944 KB
testcase_26	AC	2 ms 6,944 KB
testcase_27	AC	2 ms 6,940 KB
testcase_28	AC	1 ms 6,940 KB
testcase_29	AC	2 ms 6,940 KB
testcase_30	AC	2 ms 6,944 KB
testcase_31	AC	2 ms 6,944 KB
testcase_32	AC	2 ms 6,940 KB
testcase_33	AC	2 ms 6,940 KB
testcase_34	AC	2 ms 6,940 KB
testcase_35	AC	3 ms 6,944 KB
testcase_36	AC	4 ms 6,940 KB
testcase_37	AC	4 ms 6,940 KB
testcase_38	AC	3 ms 6,940 KB
testcase_39	AC	3 ms 6,940 KB
testcase_40	AC	3 ms 6,944 KB
testcase_41	AC	4 ms 6,948 KB
testcase_42	AC	3 ms 6,940 KB
testcase_43	AC	4 ms 6,944 KB
testcase_44	AC	3 ms 6,940 KB
testcase_45	AC	4 ms 6,940 KB
testcase_46	AC	3 ms 6,948 KB
testcase_47	AC	4 ms 6,940 KB
testcase_48	AC	3 ms 6,944 KB
testcase_49	AC	4 ms 6,944 KB
testcase_50	AC	4 ms 6,940 KB
testcase_51	AC	4 ms 6,944 KB
testcase_52	AC	4 ms 6,940 KB
testcase_53	AC	3 ms 6,944 KB
testcase_54	AC	3 ms 6,940 KB
testcase_55	AC	184 ms 20,608 KB
testcase_56	AC	38 ms 7,528 KB
testcase_57	AC	465 ms 43,504 KB
testcase_58	AC	173 ms 20,352 KB
testcase_59	AC	224 ms 19,200 KB
testcase_60	AC	152 ms 17,920 KB
testcase_61	AC	385 ms 35,712 KB
testcase_62	AC	457 ms 43,392 KB
testcase_63	AC	367 ms 32,768 KB
testcase_64	AC	172 ms 17,920 KB
testcase_65	AC	313 ms 22,016 KB
testcase_66	AC	428 ms 36,480 KB
testcase_67	AC	335 ms 32,384 KB
testcase_68	AC	226 ms 19,200 KB
testcase_69	AC	369 ms 30,720 KB
testcase_70	AC	354 ms 33,664 KB
testcase_71	AC	182 ms 21,376 KB
testcase_72	AC	461 ms 42,624 KB
testcase_73	AC	366 ms 27,008 KB
testcase_74	AC	249 ms 27,136 KB
testcase_75	AC	375 ms 30,720 KB
testcase_76	AC	295 ms 30,720 KB
testcase_77	AC	135 ms 13,696 KB
testcase_78	AC	228 ms 25,600 KB
testcase_79	AC	149 ms 17,536 KB
testcase_80	AC	440 ms 40,704 KB
testcase_81	AC	370 ms 28,672 KB
testcase_82	AC	286 ms 28,800 KB
testcase_83	AC	316 ms 28,800 KB
testcase_84	AC	203 ms 16,896 KB
testcase_85	AC	17 ms 6,940 KB
testcase_86	AC	208 ms 18,304 KB
testcase_87	AC	368 ms 35,072 KB
testcase_88	AC	113 ms 12,416 KB
testcase_89	AC	173 ms 10,240 KB
testcase_90	AC	76 ms 7,680 KB
testcase_91	AC	183 ms 10,624 KB

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

ソースコード

raw source code

#include <bits/stdc++.h>
using namespace std;

using VS = vector<string>;    using LL = long long;
using VI = vector<int>;       using VVI = vector<VI>;
using PII = pair<int, int>;   using PLL = pair<LL, LL>;
using VL = vector<LL>;        using VVL = vector<VL>;

#define ALL(a)  begin((a)),end((a))
#define RALL(a) (a).rbegin(), (a).rend()
#define PB push_back
#define EB emplace_back
#define MP make_pair
#define SZ(a) int((a).size())
#define SORT(c) sort(ALL((c)))
#define RSORT(c) sort(RALL((c)))
#define UNIQ(c) (c).erase(unique(ALL((c))), end((c)))
#define FOR(i, s, e) for (LL(i) = (s); (i) < (e); (i)++)
#define FORR(i, s, e) for (int(i) = (s); (i) > (e); (i)--)
#define debug(x) cerr << #x << ": " << x << endl
const int INF = 1e9;                          const LL LINF = 1e16;
const LL MOD = 1000000007;                    const double PI = acos(-1.0);
int DX[8] = { 0, 0, 1, -1, 1, 1, -1, -1 };    int DY[8] = { 1, -1, 0, 0, 1, -1, 1, -1 };

/* -----  2018/01/22  Problem: yukicoder409 / Link: __CONTEST_URL__  ----- */
/* ------問題------



-----問題ここまで----- */
/* -----解説等-----



----解説ここまで---- */

struct CHT2{
    typedef pair<LL,LL> PT;
    typedef LL RT;
    
    struct L{
        LL a,b;
        L(LL a, LL b): a(a),b(b){}
        bool operator<(const L &rhs) const{
            return a != rhs.a ? a > rhs.a : b < rhs.b;
        }
    };
    
    struct CP{
        LL n,d;
        L p;
        CP(LL _n,LL _d, const L &p):n(_n),d(_d),p(p){
            if(d < 0) { n *= -1; d *= -1;}
        }
        bool operator<(const CP &rhs) const{
            if(n == LINF || rhs.n == -LINF) return 0;
            if(n == -LINF || rhs.n == LINF) return 1;
            return n*rhs.d < rhs.n*d;
        }
    };
    typedef set<L>::iterator It;
    set<L> S;
    set<CP> C;
    
public:
    CHT2(){
        // 番兵
        S.insert({L(LINF,0), L(-LINF,0)});
        C.insert(cp(L(LINF,0),L(-LINF,0)));
    }
    // for debug
    void print() {
        cout << "S : "; for (auto it : S) printf("(%lld,%lld)", it.a, it.b); puts("");
        cout << "C : "; for (auto it : C) printf("(%lld,%lld)", it.n, it.d); puts("");
    }
    
    void add(LL a, LL b) {
        const L p(a,b);
        It pos = S.insert(p).first;
        if (check(*it_m1(pos), p, *it_p1(pos))) {
            // 直線(a,b)が不要
            S.erase(pos);
            return;
        }
        C.erase(cp(*it_m1(pos), *it_p1(pos)));
        {
            // 右方向の削除
            It it = it_m1(pos);
            while(it!=S.begin() && check(*it_m1(it), *it, p)) --it;
            C_erase(it, it_m1(pos));
            S.erase(++it,pos);
            pos = S.find(p);
        }
        {
            // 左方向の削除
            It it = it_p1(pos);
            while(it_p1(it)!=S.end() && check(p,*it, *it_p1(it))) ++it;
            C_erase(++pos, it);
            S.erase(pos, it);
            pos = S.find(p);
        }
        C.insert(cp(*it_m1(pos), *pos));
        C.insert(cp(*pos, *it_p1(pos)));
    }
    LL query(LL x) {
        const L &p = (--C.lower_bound(CP(x,1,L(0,0))))->p;
        return p.a*x + p.b;
    }
    
private:
    It it_p1(It a){return ++a;}
    It it_m1(It a){return --a;}
    void C_erase(It a, It b) {
        for (It it=a; it!=b; ++it) C.erase(cp(*it, *it_p1(it)));
    }
    CP cp(const L &p1, const L &p2) {
        if (p1.a == LINF) return CP(-LINF,1,p2);
        if (p2.a == -LINF) return CP(LINF,1,p2);
        return CP(p1.b-p2.b, p2.a-p1.a, p2);
    }
    bool check(const L &p1, const L &p2, const L &p3) {
        if (p1.a==p2.a && p1.b <= p2.b) return 1;
        if (p1.a == LINF || p3.a == -LINF) return 0;
        return (p2.a-p1.a)*(p3.b-p2.b) >= (p2.b-p1.b)*(p3.a-p2.a);
    }
};

LL N,A,B,W;

LL ans = 0LL;

int main() {
	cin.tie(0);
	ios_base::sync_with_stdio(false);

	cin >> N >> A >> B >> W;
	VL d(N+1);
	FOR(i,0,N){
		cin >> d[i];
	}
	CHT2 Q;
	VL dp(N+2,0);
	dp[0] = W;
	// dp(i) :=　食べてi日目を終える 
	FOR(i,0,N){ //             i日目に食べる
		Q.add(-B*i,dp[i] + B*(i*i-i)/2+ A*i); 
		dp[i+1] = B*(i*i+i)/2 + d[i] - A*i + (Q.query(i));
		//debug(dp[i]);
	}
	ans = LINF;
	FOR(i,0,N+1){ // i日目から食べてないのでストレスまみれ
		ans = min(ans,  dp[i] - A*(N-i) +B*(N-i)*(N-i+1)/2);
	}
	cout << ans << "\n";

	return 0;
}