結果

問題 No.1878 union-find の数え上げ
ユーザー 37zigen37zigen
提出日時 2021-12-24 06:15:21
言語 Java21
(openjdk 21)
結果
AC  
実行時間 764 ms / 2,000 ms
コード長 1,825 bytes
コンパイル時間 2,590 ms
コンパイル使用メモリ 87,608 KB
実行使用メモリ 79,048 KB
最終ジャッジ日時 2024-10-03 09:23:12
合計ジャッジ時間 10,297 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 634 ms
79,048 KB
testcase_01 AC 764 ms
59,276 KB
testcase_02 AC 746 ms
59,356 KB
testcase_03 AC 759 ms
55,512 KB
testcase_04 AC 721 ms
59,272 KB
testcase_05 AC 407 ms
49,884 KB
testcase_06 AC 447 ms
49,860 KB
testcase_07 AC 136 ms
41,212 KB
testcase_08 AC 136 ms
41,616 KB
testcase_09 AC 136 ms
41,136 KB
testcase_10 AC 137 ms
41,660 KB
testcase_11 AC 134 ms
41,304 KB
testcase_12 AC 135 ms
41,724 KB
testcase_13 AC 136 ms
41,344 KB
testcase_14 AC 631 ms
53,340 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.File;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Random;
import java.util.Scanner;

class Main{
	
	final long p = 998244353;
	
	long ans=1;
	
	void dfs(int v, int d, ArrayList<Integer>[] g) {
		if (d > 0) ans = ans * d % p;
		for (int u : g[v]) {
			dfs(u, d + 1, g);
		}
	}
	
	void run() throws Exception {
        Scanner sc=new Scanner(System.in);
        PrintWriter pw=new PrintWriter(System.out);
        int N=sc.nextInt();
        ArrayList<Integer>[] g=new ArrayList[N];
        for (int i=0;i<N;++i) g[i]=new ArrayList<>();
        for (int i=1;i<N;++i) {
        	int p=sc.nextInt()-1;
        	g[p].add(i);
        }
        
        dfs(0, 0, g);
        
        System.out.println(ans);
        pw.close();
	}
	
	void generate_random() throws Exception {
		Random rnd=new Random();
		for (int testcase=1;testcase<=5;++testcase) {
			File file=new File("random_"+testcase);
			PrintWriter pw=new PrintWriter(file);
			int N=100000;
			pw.println(N);
			for (int i=2;i<=N;++i) {
				int p=rnd.nextInt(1, i);
				pw.println(p);
			}			
			pw.close();
		}
	}
	
	void generate_line() throws Exception {
		File file=new File("line");
		PrintWriter pw=new PrintWriter(file);
		Random rnd=new Random();
		int N=100000;
		pw.println(N);
		for (int i=2;i<=N;++i) {
			int p=i-1;
			pw.println(p);
		}
		pw.close();
	}
	
	void generate_star() throws Exception {
		File file=new File("star");
		PrintWriter pw=new PrintWriter(file);
		Random rnd=new Random();
		int N=100000;
		pw.println(N);
		for (int i=2;i<=N;++i) {
			int p=1;
			pw.println(p);
		}
		pw.close();
	}
	
    public static void main(String args[]) throws Exception {
    	new Main().run();
    }
    
    void tr(Object...o) {
    	System.out.println(Arrays.deepToString(o));
    }
}
0