結果

問題 No.3660 LIS on Tree
コンテスト
ユーザー cho435
提出日時 2026-08-30 14:42:43
言語 C++23
(gcc 15.3.0 + boost 1.92.0)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 85 ms / 2,000 ms
+ 361µs
コード長 1,108 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 4,595 ms
コンパイル使用メモリ 377,188 KB
実行使用メモリ 22,364 KB
最終ジャッジ日時 2026-08-30 14:42:51
合計ジャッジ時間 6,320 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
#include <atcoder/all>

using namespace std;
using ll = long long;
#define rep(i, s, t) for (ll i = s; i < (ll)(t); i++)
#define all(x) begin(x), end(x)
template <class T> bool chmin(T& x, T y) {
	return x > y ? (x = y, true) : false;
}
template <class T> bool chmax(T& x, T y) {
	return x < y ? (x = y, true) : false;
}

void solve() {
	int n;
	cin >> n;
	vector<int> s(n);
	rep(i, 0, n) cin >> s[i];
	vector<vector<int>> g(n);
	rep(i, 0, n - 1) {
		int a, b;
		cin >> a >> b;
		a--, b--;
		g[a].push_back(b);
		g[b].push_back(a);
	}
	ll ans = 0;
	auto dfs = [&](auto self, int nw, int pr) -> pair<ll, ll> {
		ll mx = 0, mn = 0;
		for (int nx : g[nw]) {
			if (nx == pr) continue;
			auto [x, y] = self(self, nx, nw);
			if (s[nw] < s[nx]) chmax(mx, x);
			if (s[nw] > s[nx]) chmax(mn, y);
		}
		mx += s[nw], mn += s[nw];
		chmax(ans, max({mx, mn, mx + mn - s[nw]}));
		return {mx, mn};
	};
	dfs(dfs, 0, -1);
	cout << ans << '\n';
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout << fixed << setprecision(15);
	int t = 1;
	// cin >> t;
	while (t--) solve();
}
0