結果

問題 No.2316 Freight Train
ユーザー ks2mks2m
提出日時 2023-05-26 21:27:27
言語 Java
(openjdk 23)
結果
AC  
実行時間 707 ms / 2,000 ms
コード長 1,607 bytes
コンパイル時間 3,862 ms
コンパイル使用メモリ 76,444 KB
実行使用メモリ 77,584 KB
最終ジャッジ日時 2024-12-25 04:51:57
合計ジャッジ時間 20,009 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.PrintWriter;
public class Main {
public static void main(String[] args) throws Exception {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
String[] sa = br.readLine().split(" ");
int n = Integer.parseInt(sa[0]);
int q = Integer.parseInt(sa[1]);
sa = br.readLine().split(" ");
UnionFind uf = new UnionFind(n);
for (int i = 0; i < n; i++) {
int p = Integer.parseInt(sa[i]) - 1;
if (p != -2) {
uf.union(p, i);
}
}
PrintWriter pw = new PrintWriter(System.out);
for (int i = 0; i < q; i++) {
sa = br.readLine().split(" ");
int a = Integer.parseInt(sa[0]) - 1;
int b = Integer.parseInt(sa[1]) - 1;
if (uf.same(a, b)) {
pw.println("Yes");
} else {
pw.println("No");
}
}
pw.flush();
br.close();
}
static class UnionFind {
int[] parent, size;
int num = 0; //
UnionFind(int n) {
parent = new int[n];
size = new int[n];
num = n;
for (int i = 0; i < n; i++) {
parent[i] = i;
size[i] = 1;
}
}
void union(int x, int y) {
int px = find(x);
int py = find(y);
if (px != py) {
parent[px] = py;
size[py] += size[px];
num--;
}
}
int find(int x) {
if (parent[x] == x) {
return x;
}
parent[x] = find(parent[x]);
return parent[x];
}
/**
* xy
*/
boolean same(int x, int y) {
return find(x) == find(y);
}
/**
* x
*/
int size(int x) {
return size[find(x)];
}
}
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0