結果
問題 | No.2316 Freight Train |
ユーザー |
|
提出日時 | 2023-02-19 01:19:21 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 297 ms / 2,000 ms |
コード長 | 1,222 bytes |
コンパイル時間 | 235 ms |
コンパイル使用メモリ | 82,536 KB |
実行使用メモリ | 123,264 KB |
最終ジャッジ日時 | 2024-09-18 09:22:22 |
合計ジャッジ時間 | 9,512 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
import sys from collections import deque, Counter from math import gcd, log10, pi, sqrt, ceil from bisect import bisect, bisect_left, bisect_right, insort from typing import Iterable, TypeVar, Union, Tuple, Iterator, List import copy import heapq import itertools import math import random from fractions import Fraction from functools import lru_cache, partial, cmp_to_key import operator input = sys.stdin.readline sys.setrecursionlimit(10000000) mod = 998244353 INF = 1 << 61 DIFF = 10 ** -9 DX = [1, 0, -1, 0, 1, 1, -1, -1] DY = [0, 1, 0, -1, 1, -1, 1, -1] def read_values(): return map(int, input().split()) def read_index(): return map(lambda x: int(x) - 1, input().split()) def read_list(): return list(read_values()) def read_lists(N): return [read_list() for _ in range(N)] def main(): N, Q = read_values() P = read_list() top = [-1] * N for i, p in enumerate(P): if p == -1: top[i] = i else: top[i] = p - 1 for _ in range(N.bit_length() + 1): for i in range(N): top[i] = top[top[i]] for _ in range(Q): a, b = read_index() print("Yes" if top[a] == top[b] else "No") if __name__ == "__main__": main()