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

• 問題: No.59 鉄道の旅
• ユーザー: Ryogo1008
• 提出日時: 2017-06-05 00:24:35
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 39 ms / 5,000 ms
• コード長: 928 bytes
• コンパイル時間: 704 ms
• コンパイル使用メモリ: 63,004 KB
• 実行使用メモリ: 7,560 KB
• 最終ジャッジ日時: 2024-09-22 06:05:07
• 合計ジャッジ時間: 1,773 ms
• ジャッジサーバーID （参考情報）: judge2 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	4 ms; 7,008 KB
testcase_01	AC	4 ms; 7,008 KB
testcase_02	AC	4 ms; 6,944 KB
testcase_03	AC	4 ms; 7,008 KB
testcase_04	AC	38 ms; 7,536 KB
testcase_05	AC	4 ms; 7,084 KB
testcase_06	AC	4 ms; 7,100 KB
testcase_07	AC	4 ms; 6,996 KB
testcase_08	AC	8 ms; 7,132 KB
testcase_09	AC	7 ms; 7,192 KB
testcase_10	AC	8 ms; 7,200 KB
testcase_11	AC	6 ms; 7,112 KB
testcase_12	AC	19 ms; 7,460 KB
testcase_13	AC	39 ms; 7,496 KB
testcase_14	AC	38 ms; 7,560 KB
testcase_15	AC	4 ms; 7,108 KB

ソースコード

#include<iostream>
#include<vector>
#include<string>
using namespace std;

#define FOR(k,m,n) for(int (k)=(m);(k)<(n);(k)++)
#define rep(i,n) FOR((i),0,(n))

typedef long long ll;

const int INF=1e9+7;

class BIT{
public:
	vector<int> bit;
	int n;
	//1-indexed
	BIT(){init();}
	BIT(int n):n(n){init();}
	void init(){
		bit.clear();
		bit.resize(n+1,0);
	}
	int sum(int i){
		int s=0;
		while(i>0){
			s+=bit[i];
			i-=i&-i;
		}
		return s;
	}
	void add(int i,int x){
		while(i<=n){
			bit[i]+=x;
			i+=i&-i;
		}
	}
	int sum0(int i){
		return sum(i+1);
	}
	void add0(int i,int x){
		add(i+1,x);
	}
};

const int MAX_N=1e5+5;
const int MAX_W=1e6+6;
int w[MAX_N];

int main(){
	int n,k;
	cin>>n>>k;
	rep(i,n){
		cin>>w[i];
	}
	BIT bit(MAX_W);
	rep(i,n){
		if(w[i]>0 && bit.sum(MAX_W)-bit.sum(w[i]-1)<k){
			bit.add(w[i],1);
		}
		else if(bit.sum(-1*w[i])-bit.sum(-1*w[i]-1)>0)bit.add(-1*w[i],-1);
	}
	cout<<bit.sum(MAX_W)<<endl;
}