結果
問題 | No.2316 Freight Train |
ユーザー |
|
提出日時 | 2023-02-19 01:09:53 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,279 bytes |
コンパイル時間 | 250 ms |
コンパイル使用メモリ | 82,232 KB |
実行使用メモリ | 123,556 KB |
最終ジャッジ日時 | 2024-09-18 09:22:49 |
合計ジャッジ時間 | 8,413 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | WA * 3 |
other | AC * 5 WA * 21 |
ソースコード
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 def dfs(i): if top[i] == -1: return if P[i] == -1: top[i] = i return p = P[i] - 1 dfs(p) top[i] = top[p] for i in range(N): if top[i] != -1: continue dfs(i) for _ in range(Q): a, b = read_index() print("Yes" if top[a] == top[b] else "No") if __name__ == "__main__": main()