結果

問題 No.59 鉄道の旅
ユーザー takakintakakin
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 44 ms
12,672 KB
testcase_01 AC 39 ms
12,800 KB
testcase_02 AC 43 ms
12,544 KB
testcase_03 AC 42 ms
12,672 KB
testcase_04 AC 787 ms
26,076 KB
testcase_05 AC 46 ms
12,800 KB
testcase_06 AC 43 ms
12,800 KB
testcase_07 AC 41 ms
12,672 KB
testcase_08 AC 122 ms
13,784 KB
testcase_09 AC 108 ms
13,984 KB
testcase_10 AC 116 ms
13,568 KB
testcase_11 AC 70 ms
12,928 KB
testcase_12 AC 272 ms
13,584 KB
testcase_13 AC 764 ms
25,804 KB
testcase_14 AC 878 ms
28,732 KB
testcase_15 AC 42 ms
12,800 KB
権限があれば一括ダウンロードができます

ソースコード

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