結果

問題 No.3113 The farthest point
ユーザー moti
提出日時 2025-04-19 07:16:12
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,261 bytes
コンパイル時間 1,639 ms
コンパイル使用メモリ 146,412 KB
実行使用メモリ 18,048 KB
最終ジャッジ日時 2025-04-19 07:16:20
合計ジャッジ時間 7,011 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20 WA * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <cmath>
#include <string>
#include <map>
#include <set>

using namespace std;

#define rep(i, n) for (int i = 0; i < (n); i++)

typedef long long ll;

template<typename T>
struct Edge {
  int to;
  T cost;
};

template< typename T >
using WeightedGraph = vector<vector<Edge<T>>>;

template< typename T >
tuple<T, int, int> dfs(const WeightedGraph< T > &g, int idx, int par) {
  tuple<T, int, int> ret(0, idx, 0);
  for(auto &e : g[idx]) {
    if(e.to == par) continue;
    auto cost = dfs(g, e.to, idx);
    std::get<0>(cost) += e.cost;
    std::get<2>(cost) += 1;
    if (std::get<0>(cost) > std::get<0>(ret) || 
        (std::get<0>(cost) == std::get<0>(ret) && std::get<2>(cost) < std::get<2>(ret))) {
        ret = cost;
    }
  }
  return ret;
}

template< typename T >
T tree_diameter(const WeightedGraph< T > &g) {
  auto p = dfs(g, 0, -1);
  auto q = dfs(g, std::get<1>(p), -1);
  return std::get<0>(q);
}

signed main() {
    int N; cin >> N;
    WeightedGraph<ll> g(N);
    rep(i, N - 1) {
        int u, v; ll w; cin >> u >> v >> w;
        u--; v--;
        g[u].emplace_back(Edge<ll>{v, w});
        g[v].emplace_back(Edge<ll>{u, w});
    }
    cout << tree_diameter(g) << endl;
}
0