結果
問題 | No.872 All Tree Path |
ユーザー |
|
提出日時 | 2019-08-30 21:49:59 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
CE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 1,524 bytes |
コンパイル時間 | 746 ms |
コンパイル使用メモリ | 90,744 KB |
最終ジャッジ日時 | 2025-02-08 21:30:40 |
合計ジャッジ時間 | 2,150 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。(リジャッジ後のコンパイルエラーは公開されます)
ただし、clay言語の場合は開発者のデバッグのため、公開されます。
ただし、clay言語の場合は開発者のデバッグのため、公開されます。
コンパイルメッセージ
main.cpp:15:13: error: ‘uint32_t’ does not name a type 15 | using u32 = uint32_t; | ^~~~~~~~ main.cpp:12:1: note: ‘uint32_t’ is defined in header ‘<cstdint>’; did you forget to ‘#include <cstdint>’? 11 | #include <cmath> +++ |+#include <cstdint> 12 | main.cpp: In function ‘int main()’: main.cpp:34:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 34 | scanf("%d %d %d", &u, &v, &w); | ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
ソースコード
#include <limits> #include <iostream> #include <algorithm> #include <iomanip> #include <map> #include <set> #include <queue> #include <stack> #include <numeric> #include <bitset> #include <cmath> static const int MOD = 1000000007; using ll = long long; using u32 = uint32_t; using namespace std; template<class T> constexpr T INF = ::numeric_limits<T>::max()/32*15+208; template <typename T> struct edge { int from, to; T cost; edge(int to, T cost) : from(-1), to(to), cost(cost) {} edge(int from, int to, T cost) : from(from), to(to), cost(cost) {} }; int main() { int n; cin >> n; vector<vector<int>> G(n); vector<edge<ll>> e; for (int i = 0; i < n-1; ++i) { int u, v, w; scanf("%d %d %d", &u, &v, &w); u--; v--; G[u].emplace_back(v); G[v].emplace_back(u); e.emplace_back(u, v, w); } ll ans = 0; deque<int> Q; stack<int> s; int cnt = 0; vector<int> visited(n, 0), num(n); s.emplace(0); while(!s.empty()){ int a = s.top(); s.pop(); visited[a]++; num[a] = cnt++; Q.emplace_front(a); for (auto &&i : G[a]) { if(!visited[i]) s.emplace(i); } } vector<int> dp(n, 1); for (auto &&i : Q) { for (auto &&j : G[i]) { if(num[i] > num[j]) dp[j] += dp[i]; } } for (auto &&i : e) { ans += (ll)(n-min(dp[i.from], dp[i.to]))*min(dp[i.from], dp[i.to])*i.cost; } cout << ans*2 << "\n"; return 0; }