結果

問題 No.59 鉄道の旅
ユーザー LayCurseLayCurse
提出日時 2014-11-07 01:31:19
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 26 ms / 5,000 ms
コード長 2,717 bytes
コンパイル時間 1,714 ms
コンパイル使用メモリ 159,600 KB
実行使用メモリ 11,128 KB
最終ジャッジ日時 2024-06-07 00:35:57
合計ジャッジ時間 2,440 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 26 ms
11,128 KB
testcase_05 AC 3 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,108 KB
testcase_09 AC 10 ms
10,880 KB
testcase_10 AC 10 ms
10,240 KB
testcase_11 AC 4 ms
5,376 KB
testcase_12 AC 22 ms
5,376 KB
testcase_13 AC 15 ms
5,376 KB
testcase_14 AC 24 ms
5,376 KB
testcase_15 AC 1 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define REP(i,a,b) for(i=a;i<b;i++)
#define rep(i,n) REP(i,0,n)

#define mygc(c) (c)=getchar_unlocked()
#define mypc(c) putchar_unlocked(c)

#define ll long long
#define ull unsigned ll

void reader(int *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
void reader(int *x, int *y){reader(x);reader(y);}
void reader(int *x, int *y, int *z){reader(x);reader(y);reader(z);}
void reader(ll *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
int reader(char c[]){int i,s=0;for(;;){mygc(i);if(i!=' '&&i!='\n'&&i!='\r'&&i!='\t'&&i!=EOF) break;}c[s++]=i;for(;;){mygc(i);if(i==' '||i=='\n'||i=='\r'||i=='\t'||i==EOF) break;c[s++]=i;}c[s]='\0';return s;}

void writer(int x, char c){int s=0,m=0;char f[10];if(x<0)m=1,x=-x;while(x)f[s++]=x%10,x/=10;if(!s)f[s++]=0;if(m)mypc('-');while(s--)mypc(f[s]+'0');mypc(c);}
void writer(ll x, char c){int s=0,m=0;char f[20];if(x<0)m=1,x=-x;while(x)f[s++]=x%10,x/=10;if(!s)f[s++]=0;if(m)mypc('-');while(s--)mypc(f[s]+'0');mypc(c);}
void writer(const char c[]){int i;for(i=0;c[i]!='\0';i++)mypc(c[i]);}


template<class T>
struct fenwick{
  int size, memory;
  T *data;

  void malloc(int mem){
    memory = mem;
    data = (T*)std::malloc(sizeof(T)*mem);
  }

  void *malloc(int mem, void *workMemory){
    memory = mem;
    data = (T*)workMemory;
    return (void*)(data + mem);
  }

  void free(void){
    memory = 0;
    free(data);
  }

  void init(int N){
    int i;
    size = N;
    rep(i,N) data[i] = 0;
  }

  void add(int k, T val){
    while(k < size) data[k] += val, k |= k+1;
  }

  T get(int k){
    T res = 0;
    while(k>=0) res += data[k], k = (k&(k+1))-1;
    return res;
  }

  T range(int a, int b){
    return get(b) - get(a-1);
  }

  int kth(T k){
    int i=0, j=size, c;
    T v;
    while(i<j){
      c = (i+j)/2;
      v = get(c);
      if(v <= k) i=c+1; else j=c;
    }
    return i==size?-1:i;
  }
};

int arr[1100000];

int main(){
  int i, j, k;
  int N, K, W;
  fenwick<int> t;
  
  t.malloc(1100000);
  t.init(1100000);
  reader(&N,&K);
  while(N--){
    reader(&W);
    if(W>0){
      W = 1000000-W;
      if(t.kth(K-1) != -1 && t.kth(K-1) <= W) continue; // too slow binary search
//      if(t.get(W) >= K) continue;
      t.add(W,1); arr[W]++;
    } else {
      W = 1000000+W;
      if(arr[W]) t.add(W,-1), arr[W]--;
    }
  }
  writer(t.get(1000001),'\n');

  return 0;
}
0