結果

問題 No.59 鉄道の旅
ユーザー takakintakakin
提出日時 2020-06-08 21:34:43
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 750 ms / 5,000 ms
コード長 937 bytes
コンパイル時間 210 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 28,760 KB
最終ジャッジ日時 2024-06-07 23:44:53
合計ジャッジ時間 4,193 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
12,800 KB
testcase_01 AC 36 ms
12,672 KB
testcase_02 AC 35 ms
12,800 KB
testcase_03 AC 35 ms
12,800 KB
testcase_04 AC 693 ms
26,204 KB
testcase_05 AC 36 ms
12,800 KB
testcase_06 AC 36 ms
12,800 KB
testcase_07 AC 35 ms
12,800 KB
testcase_08 AC 104 ms
14,040 KB
testcase_09 AC 96 ms
13,988 KB
testcase_10 AC 99 ms
13,568 KB
testcase_11 AC 59 ms
12,928 KB
testcase_12 AC 244 ms
13,708 KB
testcase_13 AC 658 ms
25,700 KB
testcase_14 AC 750 ms
28,760 KB
testcase_15 AC 34 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