結果

問題 No.1424 Ultrapalindrome
ユーザー 小野寺健小野寺健
提出日時 2021-05-20 16:04:05
言語 Java21
(openjdk 21)
結果
WA  
実行時間 -
コード長 2,195 bytes
コンパイル時間 4,205 ms
コンパイル使用メモリ 80,584 KB
実行使用メモリ 56,492 KB
最終ジャッジ日時 2024-10-10 07:16:52
合計ジャッジ時間 19,741 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 118 ms
40,136 KB
testcase_01 WA -
testcase_02 AC 124 ms
40,040 KB
testcase_03 AC 131 ms
41,316 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 AC 131 ms
41,036 KB
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 AC 565 ms
48,040 KB
testcase_19 AC 590 ms
48,236 KB
testcase_20 AC 411 ms
47,700 KB
testcase_21 AC 674 ms
48,092 KB
testcase_22 AC 521 ms
48,096 KB
testcase_23 AC 309 ms
47,284 KB
testcase_24 AC 373 ms
47,684 KB
testcase_25 AC 385 ms
47,912 KB
testcase_26 AC 723 ms
51,652 KB
testcase_27 WA -
testcase_28 AC 726 ms
56,188 KB
testcase_29 WA -
testcase_30 AC 758 ms
52,612 KB
testcase_31 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.Scanner;
import java.util.List;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.PriorityQueue;
import java.util.Queue;
import java.util.Comparator;

public class No1424 {
	private static int N;
	private static HashMap<Integer, List<Integer>> Edge;
	private static List<Integer> S;

	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		N = scan.nextInt();
		Edge = new HashMap<Integer, List<Integer>>();
		for (int i=0; i < N-1; i++) {
			int v = scan.nextInt() - 1;
			int u = scan.nextInt() - 1;
			Edge.getOrDefault(v, new ArrayList<Integer>()).add(u);
			Edge.getOrDefault(u, new ArrayList<Integer>()).add(v);
		}
		scan.close();
		S = new ArrayList<Integer>();
		for (int i=0; i < N; i++) {
			if (Edge.containsKey(i) && Edge.get(i).size() == 1) {
				S.add(i);
			}
		}
		int res = -1;
		for (int i=0; i < S.size()-1; i++) {
			int v = getPath(S.get(i), i);
			if (v < 0) {
				System.out.println("No");
				return;
			} else if (res < 0) {
				res = v;
			} else if (res != v) {
				System.out.println("No");
				return;
			}
		}
		System.out.println("Yes");
	}
	
	private static class MyComparator implements Comparator {
		 
	    public int compare(Object obj1, Object obj2) {
	         
	        Integer[] num1 = (Integer[])obj1;
	        Integer[] num2 = (Integer[])obj2;
	         
	        if(num1[0] > num2[0]) {
	            return 1;
	        } else if(num1[0] < num2[0]) {
	            return -1;
	        } else{
	            return 0;
	        }
	    }
	}
	
	private static int getPath(int s, int i) {
		int[] D = new int[N];
		Arrays.fill(D, Integer.MAX_VALUE);
		D[s] = 0;
		Queue<Integer[]> q = new PriorityQueue<Integer[]>(new MyComparator());
		q.add(new Integer[] {0, s});
		while (q.size() > 0) {
			Integer[] c = q.poll();
			if (c[0] > D[c[1]]) {
				continue;
			}
			for (int p : Edge.get(c[1])) {
				if (D[p] > D[c[1]] + 1) {
					D[p] = D[c[1]] + 1;
					q.add(new Integer[] {D[p], p});
				}
			}
		}
		int res = -1;
		for (int e : S.subList(i+1, S.size())) {
			if (res < 0) {
				res = D[e];
			} else if (res != D[e]) {
				return -1;
			}
		}
		return res;
	}
}
0