結果

問題 No.871 かえるのうた
ユーザー timi
提出日時 2021-03-08 15:42:07
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 387 ms / 2,000 ms
コード長 664 bytes
コンパイル時間 99 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 34,536 KB
最終ジャッジ日時 2024-11-30 11:05:29
合計ジャッジ時間 7,282 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,K=map(int, input().split())
X=list(map(int, input().split()))
A=list(map(int, input().split()))
import bisect
D=[]
for i in range(N):
  l=bisect.bisect_left(X,X[i]-A[i])
  r=bisect.bisect_right(X,X[i]+A[i])-1
  D.append((l,r))
#print(D)
from collections import deque
d=deque()
d.append(K-1)
E=[0]*N
E[K-1]=1
ll,rr=K-1,K-1
while d:
  now=d.popleft()
  l,r=D[now]
  if 0<ll:
    if l<ll:
      for i in range(l,ll):
        if E[i]==1:
          break
        else:
          E[i]=1
          d.append(i)
  if rr<N-1:
    if rr<r:
      for i in range(r,rr,-1):
        if E[i]==1:
          break
        else:
          E[i]=1
          d.append(i)
print(sum(E))
0