結果
問題 | No.973 余興 |
ユーザー |
![]() |
提出日時 | 2019-12-06 00:14:32 |
言語 | C++17(gcc12) (gcc 12.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2,965 ms / 4,000 ms |
コード長 | 2,417 bytes |
コンパイル時間 | 13,385 ms |
コンパイル使用メモリ | 278,136 KB |
最終ジャッジ日時 | 2025-01-08 08:54:43 |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 54 |
ソースコード
#pragma GCC target("avx2") #pragma GCC optimize("O3") #include "bits/stdc++.h" using namespace std; using i64 = long long; template <typename T> struct BIT{ vector<T> elm; BIT(int n, T init = T()) : elm(n + 1, init){ } // [0, x) T sum(int x){ T val = 0; for(; x > 0; x -= x & -x) val += elm[x]; return val; } // [l, r) T sum(int l, int r){ return sum(r) - sum(l); } void add(int x, T val){ for(++x; x < elm.size(); x += x & -x) elm[x] += val; } }; signed main(){ int n, x; cin >> n >> x; vector<int> v(n); for(auto& w : v) cin >> w; vector<i64> r(n + 1, 0); for(int i = 0; i < n; ++i) r[i + 1] = r[i] + v[i]; vector<int> nex(n, n), prev(n, 0); for(int i = 0; i < n; ++i){ auto it = upper_bound(r.begin(), r.end(), r[i] + x); int idx = distance(r.begin(), it); nex[i] = min(idx, n); auto it2 = lower_bound(r.begin(), r.end(), r[i + 1] - x); idx = distance(r.begin(), it2); prev[i] = max(0, idx - 1); } /* for(auto& x : nex) cout << x << " "; cout << endl; for(auto& x : prev) cout << x << " "; cout << endl; */ vector<BIT<int>> a(n, BIT<int>(n)), b(n, BIT<int>(n)); for(int k = 1; k < n; ++k) for(int i = 0, j = k; j < n; ++i, ++j){ bool f = false; // TODO: set len // dp[i][j] : dp[i][j - x, j) // cout << j - prev[j] << " " << nex[i] - (i + 1) << " "; f |= (a[i].sum(max(i, prev[j]), j) != (j - max(i, prev[j]))); // dp[i][j] : dp[i, i + x][j, j) f |= (b[j].sum(i + 1, min(nex[i], j + 1)) != (min(nex[i], j + 1) - (i + 1))); if(f){ a[i].add(j, 1); b[j].add(i, 1); } /* cout << "[" << i << ",[" << max(i, prev[j]) << "," << j - 1 << "]] : "; cout << "[[" << i + 1 << "," << min(nex[j], j + 1) - 1 << "]," << j << "] : "; cout << "[" << i << "," << j << "] -> " << (f ? "Win" : "Lose") << endl; */ } /* for(int i = 0; i < n; ++i){ for(int j = 0; j < n; ++j) cout << a[i].sum(j, j + 1) << " \n"[j == n - 1]; } cout << endl; */ cout << (a[0].sum(n - 1, n) ? "A" : "B") << endl; }