結果

問題 No.59 鉄道の旅
ユーザー takakintakakin
提出日時 2020-06-08 21:34:43
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 588 ms / 5,000 ms
コード長 937 bytes
コンパイル時間 102 ms
コンパイル使用メモリ 10,996 KB
実行使用メモリ 26,556 KB
最終ジャッジ日時 2023-08-27 03:42:08
合計ジャッジ時間 3,564 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 25 ms
10,688 KB
testcase_01 AC 25 ms
10,584 KB
testcase_02 AC 24 ms
10,620 KB
testcase_03 AC 24 ms
10,580 KB
testcase_04 AC 540 ms
23,904 KB
testcase_05 AC 24 ms
10,640 KB
testcase_06 AC 25 ms
10,544 KB
testcase_07 AC 24 ms
10,672 KB
testcase_08 AC 80 ms
11,976 KB
testcase_09 AC 74 ms
11,952 KB
testcase_10 AC 71 ms
11,476 KB
testcase_11 AC 44 ms
10,816 KB
testcase_12 AC 192 ms
11,532 KB
testcase_13 AC 515 ms
23,588 KB
testcase_14 AC 588 ms
26,556 KB
testcase_15 AC 25 ms
10,636 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