結果

問題 No.59 鉄道の旅
ユーザー takakin
提出日時 2020-06-08 21:34:43
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 878 ms / 5,000 ms
コード長 937 bytes
コンパイル時間 320 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 28,732 KB
最終ジャッジ日時 2024-12-26 11:03:39
合計ジャッジ時間 5,130 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input=lambda: sys.stdin.readline().rstrip()
n,k=map(int,input().split())
import collections
D=collections.defaultdict(int)
W=[int(input()) for _ in range(n)]
WW=sorted(list(set([abs(i) for i in W])))
for i,ww in enumerate(WW):
  D[ww]=i+1

n_max=10**5
nn=n_max.bit_length()+1
BIT=[0]*(2**nn)
BIT.insert(0,0)

# i が与えられたとき a1~aiまでの和を計算する( O(log n) )
def bitsum(BIT,i):
  s=0
  while i>0:
    s+=BIT[i]
    i-=i&(-i)
  return s

# i と x が与えられたとき ai += x とする ( O(log n) )
def bitadd(BIT,i,x):
  while i<=2**nn:
    BIT[i]+=x
    i+=i&(-i)
  return BIT

for i,w in enumerate(W):
  ind=D[abs(w)]
  if w<0:
    if ind==1:
      if bitsum(BIT,ind)>0:
        bitadd(BIT,ind,-1)
    elif bitsum(BIT,ind)-bitsum(BIT,ind-1)>0:
      bitadd(BIT,ind,-1)
  else:
    if bitsum(BIT,n)-bitsum(BIT,ind-1)>=k:
      continue
    else:
      bitadd(BIT,ind,1)
print(bitsum(BIT,n))
0