結果
問題 |
No.3113 The farthest point
|
ユーザー |
|
提出日時 | 2025-04-19 18:15:35 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 3,381 bytes |
コンパイル時間 | 3,891 ms |
コンパイル使用メモリ | 296,552 KB |
実行使用メモリ | 33,272 KB |
最終ジャッジ日時 | 2025-04-19 18:15:44 |
合計ジャッジ時間 | 7,972 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 3 WA * 30 |
ソースコード
#line 1 "h.cpp" #include <bits/stdc++.h> #line 1 "/home/ardririy/repos/cp-libs/libraries-cpp/input.hpp" #line 7 "/home/ardririy/repos/cp-libs/libraries-cpp/input.hpp" using namespace std; vector<long long> i64_vec_IN(int n) { vector<long long> res(n); for(int i=0; i<n; i++) { cin >> res[i]; } return res; } vector<string> str_vec_IN(int n) { vector<string> res(n); for(int i=0; i<n; i++) { cin >> res[i]; } return res; } vector<vector<int>> graph_IN(int n, int m) { vector<vector<int>> g(n); int u, v; for(int i=0; i<m; i++) { cin >> u >> v; u--; v--; g[u].emplace_back(v); g[v].emplace_back(u); } return g; } vector<vector<pair<int, long long>>> weighted_graph_IN(int n, int m) { vector<vector<pair<int, long long>>> g(n); int u, v; long long w; for(int i=0; i<m; i++) { cin >> u >> v >> w; u--; v--; g[u].push_back({v, w}); g[v].push_back({u, w}); } return g; } #line 3 "h.cpp" using namespace std; // using namespace atcoder; #define all(v) v.begin(),v.end() #define resort(v) sort(v.rbegin(),v.rend()) using ll = long long; using ull = unsigned long long; using vll=vector<ll>; using vvll = vector<vector<ll>>; using P = pair<ll,ll>; using vp=vector<pair<ll, ll>>; using djks=priority_queue<P, vp, greater<P>>; const ll inf=1ll<<60; #define mod10 (ll)1e9+7 #define mod99 (ll)998244353 const double PI = acos(-1); #define rep(i,n) for (ll i=0;i<n;++i) #define per(i,n) for(ll i=n-1;i>=0;--i) #define rep2(i,a,n) for (ll i=a;i<n;++i) #define per2(i,a,n) for (ll i=n-1;i>=a;--i) template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return true; } return false; } template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return true; } return false; } ll dx[] = {1, 0, -1, 0, -1, 1, -1, 1}; ll dy[] = {0, 1, 0, -1, -1, 1, 1, -1}; // 木DPする... ll dfs1(int p, vector<vector<pair<int, ll>>>& g, vector<vp>& dists, vector<bool>& seen) { seen[p] = true; ll res = 0; for(auto [ni, w]: g[p]) { if(seen[ni]) continue; ll ret = dfs1(ni, g, dists, seen); dists[p].push_back({ni, w+ret}); chmax(res, ret); } return res; } void dfs2(int p, vector<vp>& dists, ll n_w) { if(n_w!=-inf) { dists[p].push_back({1<<30, n_w}); } sort(all(dists[p]), [](const P& a, const P& b) { return a.second > b.second; }); if(dists[p][0].first!=1<<30) { int next_w=0; rep(i, min((int)dists[p].size(), 2)){ next_w += dists[p][i].second; } dfs2(dists[p][0].first, dists, next_w); } rep2(i, 1, dists[p].size()) { if(dists[p][i].first==1<<30) continue; dfs2(dists[p][i].first, dists, dists[p][0].second + dists[p][i].second); } } void solve() { int n; cin >> n; auto g = weighted_graph_IN(n, n-1); vector<vp> dists(n); vector<bool> seen(n, false); dfs1(0, g, dists, seen); dfs2(0, dists, -inf); ll ans = 0; rep(i,n) { ll s = 0; rep(j,min((int)dists[i].size(), 2)) { s += dists[i][j].second; } chmax(ans, s); } cout << ans << '\n'; } int main() { cin.tie(0); ios::sync_with_stdio(false); int t=1; //cin >> t; while(t--)solve(); }