#244077 (C++14) No.59 鉄道の旅

• 問題: No.59 鉄道の旅
• ユーザー: ふっぴー
• 提出日時: 2018-03-14 11:53:12
• 言語: C++14 (gcc 13.2.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 44 ms / 5,000 ms
• コード長: 1,531 bytes
• コンパイル時間: 1,635 ms
• コンパイル使用メモリ: 168,680 KB
• 実行使用メモリ: 11,008 KB
• 最終ジャッジ日時: 2023-08-18 05:03:49
• 合計ジャッジ時間: 3,134 ms
• ジャッジサーバーID （参考情報）: judge14 / judge15

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	5 ms; 10,836 KB
testcase_01	AC	5 ms; 10,756 KB
testcase_02	AC	6 ms; 10,956 KB
testcase_03	AC	6 ms; 10,876 KB
testcase_04	AC	44 ms; 11,008 KB
testcase_05	AC	5 ms; 10,876 KB
testcase_06	AC	5 ms; 10,884 KB
testcase_07	AC	5 ms; 10,808 KB
testcase_08	AC	10 ms; 10,848 KB
testcase_09	AC	8 ms; 10,700 KB
testcase_10	AC	9 ms; 10,736 KB
testcase_11	AC	6 ms; 10,712 KB
testcase_12	AC	19 ms; 10,876 KB
testcase_13	AC	38 ms; 10,828 KB
testcase_14	AC	36 ms; 10,772 KB
testcase_15	AC	5 ms; 10,748 KB

ソースコード

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

#define DEBUG(x) cout<<#x<<": "<<x<<endl;
#define DEBUG_VEC(v) cout<<#v<<":";for(int i=0;i<v.size();i++) cout<<" "<<v[i]; cout<<endl

typedef long long ll;
#define vi vector<int>
#define vl vector<ll>
#define vii vector< vector<int> >
#define vll vector< vector<ll> >
#define vs vector<string>
#define pii pair<int,int>
#define pis pair<int,string>
#define psi pair<string,int>
#define pll pair<ll,ll>
#define fi first
#define se second
#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define rep1(i,n) for(int i=1;i<=(int)(n);i++)
#define all(c) c.begin(),c.end()
const int inf = 1000000001;
const ll INF = 2e18;
const ll MOD = 1000000007;
const ll mod = 1000000009;
const double pi = 3.14159265358979323846;
#define Sp(p) cout<<setprecision(15)<< fixed<<p<<endl;
int dx[4] = { 1,0,-1,0 }, dy[4] = { 0,1,0,-1 };
int dx2[8] = { 1,1,0,-1,-1,-1,0,1 }, dy2[8] = { 0,1,1,1,0,-1,-1,-1 };


class Bit {
public:
	int n;
	vl bit; // 1-index

	Bit(int _n) { n = _n; bit.resize(n + 1); }

	ll sum(int i) {
		ll s = 0;
		while (i > 0) {
			s += bit[i];
			i -= i & -i;
		}
		return s;
	}

	void add(int i, ll x) {
		while (i <= n) {
			bit[i] += x;
			i += i & -i;
		}
	}

};

int main() {
	int n, k;
	cin >> n >> k;
	Bit bit(1000005);
	int W = 1000002;
	rep(i, n) {
		int w;
		cin >> w;
		if (w > 0) {
			if (bit.sum(W) - bit.sum(w - 1) < k) {
				bit.add(w, 1);
			}
		}
		else {
			w *= -1;
			if (bit.sum(w) - bit.sum(w - 1) > 0) {
				bit.add(w, -1);
			}
		}
	}
	cout << bit.sum(W) << endl;
}