結果

問題 No.59 鉄道の旅
コンテスト
ユーザー ふっぴー
提出日時 2018-03-14 11:53:39
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 38 ms / 5,000 ms
コード長 1,531 bytes
コンパイル時間 1,407 ms
コンパイル使用メモリ 169,504 KB
実行使用メモリ 11,072 KB
最終ジャッジ日時 2024-11-27 07:49:16
合計ジャッジ時間 2,179 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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 = 1000000;
	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;
}
0