結果
問題 |
No.364 門松木
|
ユーザー |
|
提出日時 | 2016-10-09 20:10:22 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 71 ms / 3,000 ms |
コード長 | 1,674 bytes |
コンパイル時間 | 1,405 ms |
コンパイル使用メモリ | 116,012 KB |
実行使用メモリ | 15,488 KB |
最終ジャッジ日時 | 2024-11-22 00:23:20 |
合計ジャッジ時間 | 4,086 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
ソースコード
#define _USE_MATH_DEFINES #include <cstdio> #include <iostream> #include <sstream> #include <fstream> #include <iomanip> #include <algorithm> #include <cmath> #include <complex> #include <string> #include <vector> #include <list> #include <queue> #include <stack> #include <set> #include <map> #include <bitset> #include <numeric> #include <limits> #include <climits> #include <cfloat> #include <functional> #include <iterator> using namespace std; const int INF = INT_MAX / 4; vector<int> a; vector<vector<int> > edges; int solve(int curr, int prev, bool isLarge, int prevValue, int& ans) { map<int, int> maxValue; for(int next : edges[curr]){ if(next == prev) continue; int value = solve(next, curr, !isLarge, a[curr], ans); if(a[curr] == a[next] || (isLarge ^ (a[curr] < a[next]))) continue; maxValue[a[next]] = max(maxValue[a[next]], value); } int score = maxValue.size() * (maxValue.size() - 1) / 2; for(const auto& p : maxValue) score += p.second; ans = max(score, ans); maxValue[prevValue] = 0; score = maxValue.size() * (maxValue.size() - 1) / 2; for(const auto& p : maxValue) score += p.second; return score; } int main() { int n; cin >> n; a.resize(n); for(int i=0; i<n; ++i) cin >> a[i]; edges.assign(n, vector<int>()); for(int i=0; i<n-1; ++i){ int x, y; cin >> x >> y; -- x; -- y; edges[x].push_back(y); edges[y].push_back(x); } int ans = 0; solve(0, -1, true, INF, ans); solve(0, -1, false, -INF, ans); cout << ans << endl; return 0; }