結果

問題 No.59 鉄道の旅
ユーザー atn112323atn112323
提出日時 2017-01-03 22:14:09
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 45 ms / 5,000 ms
コード長 956 bytes
コンパイル時間 661 ms
コンパイル使用メモリ 61,684 KB
実行使用メモリ 11,216 KB
最終ジャッジ日時 2024-12-25 01:10:37
合計ジャッジ時間 1,810 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
7,296 KB
testcase_01 AC 3 ms
7,296 KB
testcase_02 AC 4 ms
7,296 KB
testcase_03 AC 3 ms
7,336 KB
testcase_04 AC 45 ms
11,136 KB
testcase_05 AC 4 ms
7,168 KB
testcase_06 AC 4 ms
7,296 KB
testcase_07 AC 5 ms
7,424 KB
testcase_08 AC 9 ms
11,216 KB
testcase_09 AC 10 ms
10,880 KB
testcase_10 AC 12 ms
10,608 KB
testcase_11 AC 5 ms
7,296 KB
testcase_12 AC 17 ms
7,296 KB
testcase_13 AC 36 ms
7,452 KB
testcase_14 AC 37 ms
7,680 KB
testcase_15 AC 4 ms
7,168 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>

using namespace std;

template <typename T>
class BinaryIndexTree {
public:
	explicit BinaryIndexTree(int size) {
		size_ = 1;
		while (size_ < size) {
			size_ <<= 1;
		}
		tree_.resize(size_);
	}
	void add(int k, const T& x) {
		k++;
		while (k <= size_) {
			tree_[k - 1] += x;
			k += k & -k;
		}
	}
	T sum(int k) {
		k++;
		T s = 0;
		while (k > 0) {
			s += tree_[k - 1];
			k -= k & -k;
		}
		return s;
	}

private:
	int size_;
	vector<T> tree_;
};

const int MAXN = 100000;
const int MAXW = 1000000;

int N, K, W;
int picked[MAXW + 1];

int main() {
	BinaryIndexTree<int> bit(MAXW + 1);
	int ans = 0;
	cin >> N >> K;
	for (int i = 0; i < N; i++) {
		cin >> W;
		if (W > 0) {
			int cnt = bit.sum(MAXW) - bit.sum(W - 1);
			if (cnt < K) {
				ans++;
				bit.add(W, 1);
				picked[W]++;
			}
		} else if (picked[-W] > 0) {
			ans--;
			bit.add(-W, -1);
			picked[-W]--;
		}
	}
	cout << ans << endl;
	return 0;
}
0