結果
問題 | No.871 かえるのうた |
ユーザー |
![]() |
提出日時 | 2020-05-08 14:41:37 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 284 ms / 2,000 ms |
コード長 | 1,328 bytes |
コンパイル時間 | 136 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 28,888 KB |
最終ジャッジ日時 | 2024-11-30 11:01:30 |
合計ジャッジ時間 | 5,372 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 49 |
ソースコード
import sysdef input(): return sys.stdin.readline().strip()def list2d(a, b, c): return [[c] * b for i in range(a)]def list3d(a, b, c, d): return [[[d] * c for j in range(b)] for i in range(a)]def list4d(a, b, c, d, e): return [[[[e] * d for j in range(c)] for j in range(b)] for i in range(a)]def ceil(x, y=1): return int(-(-x // y))def INT(): return int(input())def MAP(): return map(int, input().split())def LIST(N=None): return list(MAP()) if N is None else [INT() for i in range(N)]def Yes(): print('Yes')def No(): print('No')def YES(): print('YES')def NO(): print('NO')sys.setrecursionlimit(10 ** 9)INF = 10 ** 18MOD = 10 ** 9 + 7EPS = 10 ** -10def bfs(src):from collections import dequefrom bisect import bisect_left, bisect_rightque = deque([src])visited = [0] * Nvisited[src] = 1while que:u = que.popleft()l = bisect_left(X, X[u]-A[u])r = bisect_right(X, X[u]+A[u]) - 1for v in range(l, u):if visited[v]:breakvisited[v] = 1que.append((v))for v in range(r, u, -1):if visited[v]:breakvisited[v] = 1que.append((v))return visitedN, K = MAP()K -= 1X = LIST()A = LIST()res = bfs(K)ans = sum(res)print(ans)