結果

問題 No.973 余興
ユーザー kimiyuki
提出日時 2020-01-18 06:50:11
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,837 ms / 4,000 ms
コード長 2,140 bytes
コンパイル時間 2,856 ms
コンパイル使用メモリ 199,652 KB
最終ジャッジ日時 2025-01-08 19:54:50
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 54
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define REP(i, n) for (int i = 0; (i) < (int)(n); ++ (i))
#define REP3(i, m, n) for (int i = (m); (i) < (int)(n); ++ (i))
#define REP_R(i, n) for (int i = (int)(n) - 1; (i) >= 0; -- (i))
#define REP3R(i, m, n) for (int i = (int)(n) - 1; (i) >= (int)(m); -- (i))
#define ALL(x) std::begin(x), std::end(x)
using namespace std;
template <typename X, typename T> auto make_table(X x, T a) { return vector<T>(x, a); }
template <typename X, typename Y, typename Z, typename... Zs> auto make_table(X x, Y y, Z z, Zs... zs) { auto cont = make_table(y, z, zs...); return vector<decltype(cont)>(x, cont); }
template <typename T> istream & operator >> (istream & in, vector<T> & xs) { REP (i, xs.size()) { in >> xs[i]; } return in; }

/**
 * @brief a binary search / 二分探索
 * @param[in] p  a monotone predicate defined on $[l, r)$
 * @return $\min \lbrace x \in [l, r) \mid p(x) \rbrace$, or r if it doesn't exist
 */
template <typename UnaryPredicate>
int64_t binsearch(int64_t l, int64_t r, UnaryPredicate p) {
    assert (l <= r);
    -- l;
    while (r - l > 1) {
        int64_t m = l + (r - l) / 2;
        (p(m) ? r : l) = m;
    }
    return r;
}


bool solve(int n, int x, const vector<int64_t> & a) {
    vector<int64_t> acc(n + 1);
    partial_sum(ALL(a), acc.begin() + 1);
    auto hr = make_table(n + 1, n + 2, int16_t());
    auto vr = make_table(n + 1, n + 2, int16_t());
    REP (len, n + 1) REP (l, n - len + 1) {
        int r = l + len;
        bool cur;
        if (len == 0) {
            cur = true;
        } else {
           int m1 = binsearch(l + 1, r, [&](int m1) {
               return acc[m1] - acc[l] > x;
           });
           int m2 = binsearch(l, r, [&](int m2) {
               return acc[r] - acc[m2] <= x;
           });
           cur = vr[r][l + 1] - vr[r][m1] + hr[l][r] - hr[l][m2];
        }
        hr[l][r + 1] = hr[l][r] + not cur;
        vr[r][l] = vr[r][l + 1] + not cur;
    }
    return not (hr[0][n + 1] - hr[0][n]);
}

int main() {
    int n, x; cin >> n >> x;
    vector<int64_t> a(n); cin >> a;
    cout << (solve(n, x, a) ? "A" : "B") << endl;
    return 0;
}
0