結果

問題 No.364 門松木
ユーザー 37zigen37zigen
提出日時 2016-06-11 09:55:14
言語 Java21
(openjdk 21)
結果
AC  
実行時間 1,425 ms / 3,000 ms
コード長 2,414 bytes
コンパイル時間 2,464 ms
コンパイル使用メモリ 79,436 KB
実行使用メモリ 359,880 KB
最終ジャッジ日時 2024-10-09 12:50:26
合計ジャッジ時間 40,229 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 407 ms
241,304 KB
testcase_01 AC 401 ms
241,240 KB
testcase_02 AC 387 ms
241,468 KB
testcase_03 AC 403 ms
241,336 KB
testcase_04 AC 402 ms
241,280 KB
testcase_05 AC 398 ms
241,176 KB
testcase_06 AC 408 ms
241,396 KB
testcase_07 AC 381 ms
241,252 KB
testcase_08 AC 1,365 ms
346,504 KB
testcase_09 AC 1,345 ms
346,360 KB
testcase_10 AC 1,309 ms
347,672 KB
testcase_11 AC 1,300 ms
344,808 KB
testcase_12 AC 1,286 ms
347,836 KB
testcase_13 AC 1,331 ms
345,372 KB
testcase_14 AC 1,302 ms
345,892 KB
testcase_15 AC 1,244 ms
344,512 KB
testcase_16 AC 1,309 ms
345,796 KB
testcase_17 AC 1,351 ms
346,516 KB
testcase_18 AC 1,299 ms
346,432 KB
testcase_19 AC 1,293 ms
345,192 KB
testcase_20 AC 1,268 ms
345,644 KB
testcase_21 AC 1,278 ms
345,872 KB
testcase_22 AC 1,189 ms
344,944 KB
testcase_23 AC 1,297 ms
346,748 KB
testcase_24 AC 1,293 ms
344,688 KB
testcase_25 AC 1,394 ms
348,008 KB
testcase_26 AC 1,425 ms
346,592 KB
testcase_27 AC 1,376 ms
359,880 KB
testcase_28 AC 1,275 ms
345,956 KB
testcase_29 AC 1,347 ms
345,408 KB
testcase_30 AC 1,265 ms
348,084 KB
testcase_31 AC 401 ms
241,432 KB
testcase_32 AC 405 ms
241,480 KB
testcase_33 AC 1,379 ms
357,620 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

package yukicoder;

import java.util.ArrayList;
import java.util.Scanner;

public class Main{
	public static void main(String[] args) {
		new Main().solve();
	}

	ArrayList<Integer>[] edges = new ArrayList[50000];
	int[] A;
	// dp[v][A[u]]:頂点vの子頂点uをvにつないだ時の門松列の総数
	int[][] dp = new int[50001][1001];
	int[] num_up = new int[50001];// num_up[v]:頂点vの持つ数字より大きい数を持つ子頂点の数
	int[] num_down = new int[50001];// num_up[v]:頂点vの持つ数字より小さい数を持つ子頂点の数
	int[] sc_down = new int[50001];// sc_up[v]:頂点vの子頂点がvより小さくなるような場合に含める最大の門松列の総数
	int[] sc_up = new int[50001];// sc_down[v]:頂点vの子頂点がvより大きくなるな場合に含める最大の門松列の総数

	int dfs(int cur, int pre) {
		int ret = 0;
		for (int i = 0; i < edges[cur].size(); i++) {
			int u = edges[cur].get(i);
			if (pre != u) {
				ret = Math.max(ret, dfs(u, cur));
				if (A[cur] > A[u]) {
					if (dp[u][A[cur]] >= 0) {
						dp[cur][A[u]] = Math.max(sc_up[u] - dp[u][A[cur]], dp[cur][A[u]]);
					} else {
						dp[cur][A[u]] = Math.max(num_up[u] + sc_up[u], dp[cur][A[u]]);
					}
				} else if (A[cur] < A[u]) {
					if (dp[u][A[cur]] >= 0) {
						dp[cur][A[u]] = Math.max(sc_down[u] - dp[u][A[cur]], dp[cur][A[cur]]);
					} else {
						dp[cur][A[u]] = Math.max(num_down[u] + sc_down[u], dp[cur][A[u]]);
					}
				}
			}
		}
		for (int i = 1; i <= 1000; i++) {
			if (dp[cur][i] >= 0) {
				if (A[cur] > i) {
					num_down[cur]++;
					sc_down[cur] += dp[cur][i];
				} else if (A[cur] < i) {
					num_up[cur]++;
					sc_up[cur] += dp[cur][i];
				}
			}
		}
		sc_down[cur] += num_down[cur] * (num_down[cur] - 1) / 2;
		sc_up[cur] += num_up[cur] * (num_up[cur] - 1) / 2;
		return Math.max(ret, Math.max(sc_down[cur], sc_up[cur]));
	}

	int N;

	void solve() {
		Scanner sc = new Scanner(System.in);
		N = sc.nextInt();
		A = new int[N];
		for (int i = 0; i < 50000; i++)
			edges[i] = new ArrayList<>();
		for (int i = 0; i < dp.length; i++)
			for (int j = 0; j < dp[0].length; j++)
				dp[i][j] = -1;
		for (int i = 0; i < N; i++)
			A[i] = sc.nextInt();
		for (int i = 0; i < N - 1; i++) {
			int x = sc.nextInt() - 1;
			int y = sc.nextInt() - 1;
			edges[x].add(y);
			edges[y].add(x);
		}
		int ret = dfs(0, -1);
		System.out.println(ret);
	}
}
0