結果

問題 No.364 門松木
ユーザー 37zigen37zigen
提出日時 2016-06-11 11:43:55
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 160 ms / 3,000 ms
コード長 2,120 bytes
コンパイル時間 554 ms
コンパイル使用メモリ 64,556 KB
実行使用メモリ 208,196 KB
最終ジャッジ日時 2024-04-17 18:44:13
合計ジャッジ時間 5,619 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 51 ms
200,036 KB
testcase_01 AC 53 ms
200,068 KB
testcase_02 AC 50 ms
199,960 KB
testcase_03 AC 52 ms
200,176 KB
testcase_04 AC 50 ms
200,168 KB
testcase_05 AC 50 ms
200,232 KB
testcase_06 AC 48 ms
200,092 KB
testcase_07 AC 50 ms
200,108 KB
testcase_08 AC 153 ms
202,688 KB
testcase_09 AC 144 ms
202,716 KB
testcase_10 AC 133 ms
202,844 KB
testcase_11 AC 146 ms
202,768 KB
testcase_12 AC 140 ms
202,616 KB
testcase_13 AC 125 ms
202,620 KB
testcase_14 AC 130 ms
202,556 KB
testcase_15 AC 131 ms
202,640 KB
testcase_16 AC 130 ms
202,544 KB
testcase_17 AC 151 ms
202,724 KB
testcase_18 AC 147 ms
202,836 KB
testcase_19 AC 148 ms
203,068 KB
testcase_20 AC 141 ms
202,828 KB
testcase_21 AC 152 ms
202,812 KB
testcase_22 AC 115 ms
202,328 KB
testcase_23 AC 157 ms
203,924 KB
testcase_24 AC 151 ms
202,460 KB
testcase_25 AC 160 ms
204,072 KB
testcase_26 AC 157 ms
203,548 KB
testcase_27 AC 148 ms
202,940 KB
testcase_28 AC 150 ms
202,460 KB
testcase_29 AC 151 ms
203,808 KB
testcase_30 AC 148 ms
204,212 KB
testcase_31 AC 50 ms
200,120 KB
testcase_32 AC 51 ms
200,308 KB
testcase_33 AC 148 ms
208,196 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

// ConsoleApplication8.cpp : コンソール アプリケーションのエントリ ポイントを定義します。
//


#include <iostream>
#include <vector>
#include <algorithm>
#include <string.h>
using namespace std;

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

int dfs(int cur, int pre) {
	int ret = 0;
	for (auto& u:edges[cur]) {
			if (pre != u) {
				ret = max(ret, dfs(u, cur));
				if (A[cur] > A[u]) {
					if (dp[u][A[cur]] >= 0) {
						dp[cur][A[u]] = max(sc_up[u] - dp[u][A[cur]], dp[cur][A[u]]);
					}
					else {
						dp[cur][A[u]] = 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]] = max(sc_down[u] - dp[u][A[cur]], dp[cur][A[cur]]);
					}
					else {
						dp[cur][A[u]] = 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 max(ret, max(sc_down[cur], sc_up[cur]));
	}

	int N;

void solve() {
	int x, y;
	cin>>N;
	memset(dp, -1, sizeof(dp));
	for (int i = 0; i < N; i++)
		cin >> A[i];
	for (int i = 0; i < N - 1; i++) {
		cin >> x >> y;
		edges[x-1].push_back(y-1);
		edges[y-1].push_back(x-1);
	}
	int ret = dfs(0, -1);
	cout<<ret;
}
int main()
{
	solve();
	return 0;
}
0