結果

問題 No.59 鉄道の旅
ユーザー tkmst201tkmst201
提出日時 2017-05-18 23:27:37
言語 C++11
(gcc 13.3.0)
結果
RE  
実行時間 -
コード長 1,449 bytes
コンパイル時間 1,213 ms
コンパイル使用メモリ 160,904 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-09-18 19:16:03
合計ジャッジ時間 2,557 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 4 WA * 4 RE * 4
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:66:24: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   66 |         REP(i, n) scanf("%d", w + i);
      |                   ~~~~~^~~~~~~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

#define FOR(i,a,b) for(int i=(a);i<(b);i++)
#define REP(i,n) FOR(i,0,n)
#define ALL(v) (v).begin(),(v).end()

template<typename A, typename B> inline bool chmax(A &a, B b) { if (a<b) { a=b; return 1; } return 0; }
template<typename A, typename B> inline bool chmin(A &a, B b) { if (a>b) { a=b; return 1; } return 0; }

typedef unsigned long long ull;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<pii, pii> P;

const ll INF = 1ll<<29;
const ll MOD = 1000000007;
const double EPS  = 1e-10;

template<typename T> struct BIT {
	vector<T> bit;
	int n;
	
	BIT() {}
	BIT(int n) : bit(vector<T>(n + 2, 0)), n(n) {}
	
	T sum(int i) {
		i++;
		T s = 0;
		while (i > 0) {
			s += bit[i];
			i -= i & -i;
		}
		return s;
	}
	
	void add(int i, T x) {
		i++;
		while (i <= n) {
			bit[i] += x;
			i += i & -i;
		}
	}
	
	int lower_bound(T x) {
		int l = -1, r = bit.size();
		while (r - l > 1) {
			int m = (l + r) / 2;
			
			if (sum(m) >= x) r = m;
			else l = m;
		}
		if (r == bit.size()) r = -1;
		
		return r;
	}
};

int n, k;
int w[112345];

int main() {
	cin >> n >> k;
	REP(i, n) scanf("%d", w + i);
	
	BIT<int> bit(112345);
	
	REP(i, n) {
		if (w[i] > 0) {
			if (bit.sum(112345) - bit.sum(w[i] - 1) < k) bit.add(w[i], 1);
		}
		else {
			if (bit.sum(-w[i]) - bit.sum(-w[i] - 1) > 0) bit.add(-w[i], -1);
		}
	}
	
	cout << bit.sum(112345) << endl;
	
	return 0;
}
0