yukicoder No.59 - 鉄道の旅

Source

ニコニコミュニティ
問題文

問題概要

$N$ 個の駅がある.
各駅 $i$ について,$0$ でない整数 $W_i$ が与えられ,$W_i$ が正ならば,その駅で重さ $|W_i|$ 荷物を $1$ つ積む,$W_i$ が負であれば,その駅で,重さ $|W_i|$ の荷物を $1$ つ降ろすことを意味する.
ただし,荷物を積む際は,その荷物の重さ以上の重さの荷物が既に $K$ 個以上積まれている状態であれば積まない.
また,荷物を降ろす際,該当する重さの荷物が積まれていなければ降ろさない,とする.
最初荷物を積んでいないとして,この $N$ 個の駅を順番に訪れた後の積み荷の個数を求める問題.

解法

シミュレーションすれば良い.
ただし,ある重さ以上の荷物が何個積まれているかを高速に求める必要があるので,Fenwick Treeなどを用いる.
時間計算量は $O(N \log |W_k|)$ 程度.

C++によるスパゲッティなソースコード

#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)

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 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);}

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){size=N;memset(data,0,sizeof(T)*N);}
  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){if(b==-1)b=size-1;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.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;
}

Current time: 2017年07月24日15時46分53秒
Last modified: 2014年12月23日23時10分37秒 (by laycrs)
Tags: Competitive_Programming yukicoder
トップページに戻る

Logged in as: unknown user (not login)

ログイン: