結果
問題 | No.59 鉄道の旅 |
ユーザー | kimiyuki |
提出日時 | 2016-11-22 21:03:21 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 41 ms / 5,000 ms |
コード長 | 1,600 bytes |
コンパイル時間 | 1,580 ms |
コンパイル使用メモリ | 78,260 KB |
実行使用メモリ | 7,780 KB |
最終ジャッジ日時 | 2023-08-26 07:43:12 |
合計ジャッジ時間 | 1,711 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,376 KB |
testcase_01 | AC | 2 ms
4,380 KB |
testcase_02 | AC | 2 ms
4,376 KB |
testcase_03 | AC | 1 ms
4,376 KB |
testcase_04 | AC | 41 ms
7,160 KB |
testcase_05 | AC | 3 ms
6,944 KB |
testcase_06 | AC | 4 ms
6,948 KB |
testcase_07 | AC | 3 ms
7,232 KB |
testcase_08 | AC | 7 ms
6,936 KB |
testcase_09 | AC | 8 ms
6,948 KB |
testcase_10 | AC | 7 ms
6,944 KB |
testcase_11 | AC | 3 ms
4,376 KB |
testcase_12 | AC | 16 ms
4,380 KB |
testcase_13 | AC | 40 ms
7,780 KB |
testcase_14 | AC | 39 ms
7,216 KB |
testcase_15 | AC | 4 ms
7,004 KB |
ソースコード
#include <iostream> #include <vector> #include <algorithm> #include <functional> #define repeat(i,n) for (int i = 0; (i) < (n); ++(i)) #define whole(f,x,...) ([&](decltype((x)) whole) { return (f)(begin(whole), end(whole), ## __VA_ARGS__); })(x) using namespace std; template <typename T> struct binary_indexed_tree { // on monoid vector<T> data; T unit; function<T (T,T)> append; // associative template <typename F> binary_indexed_tree(size_t n, T a_unit, F a_append) : data(n, a_unit), unit(a_unit), append(a_append) {} void point_append(size_t i, T w) { // data[i] += w for (size_t j = i+1; j <= data.size(); j += j & -j) data[j-1] = append(data[j-1], w); } int initial_range_concat(size_t i) { // sum [0, i) T acc = unit; for (size_t j = i; 0 < j; j -= j & -j) acc = append(acc, data[j-1]); return acc; } T point_get(size_t i) { return initial_range_concat(i+1) - initial_range_concat(i); } }; int main() { int n, k; cin >> n >> k; vector<int> ws(n); repeat (i,n) cin >> ws[i]; int w_max = max(*whole(max_element, ws), - *whole(min_element, ws)); binary_indexed_tree<int> bit(w_max+1, int(), plus<int>()); for (int w : ws) { int i = w_max - abs(w); if (w > 0) { if (bit.initial_range_concat(i+1) < k) { bit.point_append(i, 1); } } else if (w < 0) { if (bit.point_get(i)) { bit.point_append(i, -1); } } } cout << bit.initial_range_concat(w_max) << endl; return 0; }