結果
| 問題 | 
                            No.2316 Freight Train
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 2023-05-26 21:25:04 | 
| 言語 | Python3  (3.13.1 + numpy 2.2.1 + scipy 1.14.1)  | 
                    
| 結果 | 
                             
                                WA
                                 
                             
                            
                         | 
                    
| 実行時間 | - | 
| コード長 | 430 bytes | 
| コンパイル時間 | 122 ms | 
| コンパイル使用メモリ | 12,416 KB | 
| 実行使用メモリ | 30,960 KB | 
| 最終ジャッジ日時 | 2024-12-25 04:39:18 | 
| 合計ジャッジ時間 | 31,188 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge2 / judge3 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 2 WA * 1 | 
| other | AC * 6 WA * 20 | 
ソースコード
N, Q = map(int, input().split())
P = list(map(int, input().split()))
train_line = [0] * N
line_num = 1
for i in range(N):
    if P[i] == -1:
        train_line[i] = line_num
        line_num += 1
for i in range(N):
    if P[i] != -1:
        train_line[i] = train_line[P[i]-1]
for _ in range(Q):
    A, B = map(int, input().split())
    if train_line[A-1] == train_line[B-1]:
        print("Yes")
    else:
        print("No")