結果
問題 | No.59 鉄道の旅 |
ユーザー |
|
提出日時 | 2021-02-01 02:33:42 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 2,426 bytes |
コンパイル時間 | 1,082 ms |
コンパイル使用メモリ | 131,928 KB |
最終ジャッジ日時 | 2025-01-18 10:33:32 |
ジャッジサーバーID (参考情報) |
judge4 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 6 WA * 6 |
ソースコード
#include <iostream>#include <vector>#include <algorithm>#include <cmath>#include <queue>#include <string>#include <map>#include <set>#include <stack>#include <tuple>#include <deque>#include <array>#include <numeric>#include <bitset>#include <iomanip>#include <cassert>#include <chrono>#include <random>#include <limits>#include <iterator>#include <functional>#include <sstream>#include <fstream>#include <complex>#include <cstring>#include <unordered_map>#include <unordered_set>using namespace std;using ll = long long;constexpr int INF = 1001001001;constexpr int mod = 1000000007;// constexpr int mod = 998244353;template<class T>inline bool chmax(T& x, T y){if(x < y){x = y;return true;}return false;}template<class T>inline bool chmin(T& x, T y){if(x > y){x = y;return true;}return false;}template<typename T>struct BinaryIndexedTree{vector<T> data;int sz;BinaryIndexedTree(int n){sz = n + 2;data.assign(sz, 0);}void add(int i, T x){++i;while(i < sz){data[i] += x;i += i & -i;}}// [0, i] の区間和T sum(int i){T res = 0;++i;while(i > 0){res += data[i];i -= i & -i;}return res;}// [l, r) の区間和T sum(int l, int r){if(l > r) return 0;return sum(r - 1) - sum(l - 1);}// val 以上を満たす 0-indexed の位置を返すint lower_bound(T val){// if(val <= 0) return -1;int res = 0;int n = 1;while(n < sz) n <<= 1;for(int k = n >> 1; k > 0; k >>= 1){if(res + k < sz && data[res + k] < val){val -= data[res + k];res += k;}}return res; // 1-indexed であれば、res + 1 を返す}};int main(){ios::sync_with_stdio(false);cin.tie(nullptr);constexpr int MAX = 1000001;int N, K, W;cin >> N >> K;BinaryIndexedTree<int> bit(MAX);for(int i = 0; i < N; ++i){cin >> W;if(W > 0){if(bit.sum(W, MAX) < K) bit.add(W, 1);}else{bit.add(-W, -(bit.sum(-W) - bit.sum(0, -W)));}}cout << bit.sum(0, MAX) << endl;return 0;}