#19695 (C++11) No.59 鉄道の旅

• 問題: No.59 鉄道の旅
• ユーザー: 古寺いろは
• 提出日時: 2015-04-13 22:35:29
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 48 ms / 5,000 ms
• コード長: 778 bytes
• コンパイル時間: 1,683 ms
• コンパイル使用メモリ: 157,632 KB
• 実行使用メモリ: 11,136 KB
• 最終ジャッジ日時: 2024-06-07 00:55:40
• 合計ジャッジ時間: 2,190 ms
• ジャッジサーバーID （参考情報）: judge5 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	3 ms; 5,376 KB
testcase_04	AC	48 ms; 11,136 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	11 ms; 11,008 KB
testcase_09	AC	13 ms; 10,752 KB
testcase_10	AC	12 ms; 10,112 KB
testcase_11	AC	4 ms; 5,376 KB
testcase_12	AC	17 ms; 5,376 KB
testcase_13	AC	36 ms; 5,376 KB
testcase_14	AC	36 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB

ソースコード

#include "bits/stdc++.h"
using namespace std;
#define MAX 20

class BIT{
public:
	int dp[1 << MAX];
	void init(){
		for (int i = 0; i < (1 << MAX); i++)
		{
			dp[i] = 0;
		}
	}

	long long bitcall(int a){
		int ret = 0;
		while (a > 0){
			ret += dp[a];
			a &= a - 1;
		}
		return ret;
	}

	void bitadd(int a, int b){
		while (a < (1 << MAX)){
			dp[a] += b;
			a += a - (a & (a - 1));
		}
	}
};

int num[1 << MAX];
BIT bit;

int main(){
	int N, K;
	cin >> N >> K;
	int count = 0;
	for (int i = 0; i < N; i++)
	{
		int W;
		cin >> W;
		if (W < 0){
			W = -W;
			if (num[W] == 0) continue;
			num[W]--;
			bit.bitadd(W, -1);
			count--;
		}
		else{
			if (count - bit.bitcall(W - 1) >= K) continue;
			num[W]++;
			bit.bitadd(W, 1);
			count++;
		}
	}
	cout << count << endl;
}