結果

問題 No.871 かえるのうた
ユーザー stng
提出日時 2022-07-23 17:22:37
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 831 bytes
コンパイル時間 240 ms
コンパイル使用メモリ 82,452 KB
実行使用メモリ 92,520 KB
最終ジャッジ日時 2024-07-05 06:47:10
合計ジャッジ時間 5,693 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 22 WA * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque
import bisect

n,k = map(int,input().split())
x = [int(i) for i in input().split()]
a = [int(i) for i in input().split()]

ans = [0]*(n)

d = deque()
d.append(k-1)
l = k-1
r = k-1
ans[k-1] = 1

while len(d):
    tmp = d.popleft()
    pos = x[tmp]
    nl = bisect.bisect_left(x,pos-a[tmp])
    nr = bisect.bisect_left(x,pos+a[tmp])
    #print(nl,nr)
    #exit()
    if nl < l:
        for i in range(nl,l):
            #print(i,"ll")
            if ans[i] == 1:
                break
            d.append(i)
            ans[i] = 1
    if nr > r:
        for i in range(nr-1,r,-1):
            if i >= n:
                continue
            #print(i,"rr")
            if ans[i] == 1:
                break
            d.append(i)
            ans[i] = 1
    #exit()
    l = nl
    r = nr
print(sum(ans))
0