結果
問題 | No.1878 union-find の数え上げ |
ユーザー | tenten |
提出日時 | 2022-04-06 11:05:20 |
言語 | Java21 (openjdk 21) |
結果 |
AC
|
実行時間 | 332 ms / 2,000 ms |
コード長 | 1,567 bytes |
コンパイル時間 | 2,332 ms |
コンパイル使用メモリ | 75,332 KB |
実行使用メモリ | 80,468 KB |
最終ジャッジ日時 | 2023-08-18 12:43:44 |
合計ジャッジ時間 | 6,761 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 281 ms
80,468 KB |
testcase_01 | AC | 322 ms
63,760 KB |
testcase_02 | AC | 332 ms
63,056 KB |
testcase_03 | AC | 324 ms
63,040 KB |
testcase_04 | AC | 312 ms
65,020 KB |
testcase_05 | AC | 161 ms
55,324 KB |
testcase_06 | AC | 162 ms
55,256 KB |
testcase_07 | AC | 44 ms
49,520 KB |
testcase_08 | AC | 44 ms
47,444 KB |
testcase_09 | AC | 44 ms
47,328 KB |
testcase_10 | AC | 46 ms
49,428 KB |
testcase_11 | AC | 47 ms
49,468 KB |
testcase_12 | AC | 43 ms
49,640 KB |
testcase_13 | AC | 45 ms
49,364 KB |
testcase_14 | AC | 211 ms
60,844 KB |
ソースコード
import java.io.*; import java.util.*; public class Main { static final int MOD = 998244353; static ArrayList<ArrayList<Integer>> graph = new ArrayList<>(); static int[] depths; public static void main(String[] args) throws Exception { Scanner sc = new Scanner(); int n = sc.nextInt(); for (int i = 0; i < n; i++) { graph.add(new ArrayList<>()); } for (int i = 1; i < n; i++) { graph.get(sc.nextInt() - 1).add(i); } depths = new int[n]; setDepth(0, 0); long ans = 1; for (int i = 1; i < n; i++) { ans *= depths[i]; ans %= MOD; } System.out.println(ans); } static void setDepth(int idx, int d) { depths[idx] = d; for (int x : graph.get(idx)) { setDepth(x, d + 1); } } } class Scanner { BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); StringTokenizer st = new StringTokenizer(""); public Scanner() throws Exception { } public int nextInt() throws Exception { return Integer.parseInt(next()); } public long nextLong() throws Exception { return Long.parseLong(next()); } public double nextDouble() throws Exception { return Double.parseDouble(next()); } public String next() throws Exception { if (!st.hasMoreTokens()) { st = new StringTokenizer(br.readLine()); } return st.nextToken(); } }