結果
問題 | No.872 All Tree Path |
ユーザー | siman |
提出日時 | 2021-08-02 18:37:33 |
言語 | C++17(clang) (17.0.6 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 315 ms / 3,000 ms |
コード長 | 1,203 bytes |
コンパイル時間 | 1,289 ms |
コンパイル使用メモリ | 106,840 KB |
実行使用メモリ | 47,180 KB |
最終ジャッジ日時 | 2023-10-14 20:51:06 |
合計ジャッジ時間 | 4,845 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 281 ms
26,756 KB |
testcase_01 | AC | 296 ms
26,676 KB |
testcase_02 | AC | 315 ms
26,696 KB |
testcase_03 | AC | 218 ms
47,180 KB |
testcase_04 | AC | 5 ms
12,768 KB |
testcase_05 | AC | 275 ms
26,756 KB |
testcase_06 | AC | 288 ms
26,836 KB |
testcase_07 | AC | 272 ms
26,784 KB |
testcase_08 | AC | 25 ms
14,132 KB |
testcase_09 | AC | 25 ms
14,152 KB |
testcase_10 | AC | 25 ms
14,272 KB |
testcase_11 | AC | 26 ms
14,192 KB |
testcase_12 | AC | 26 ms
14,184 KB |
testcase_13 | AC | 5 ms
12,888 KB |
testcase_14 | AC | 5 ms
12,764 KB |
testcase_15 | AC | 6 ms
12,736 KB |
testcase_16 | AC | 5 ms
12,768 KB |
testcase_17 | AC | 5 ms
12,856 KB |
testcase_18 | AC | 5 ms
12,772 KB |
testcase_19 | AC | 4 ms
12,748 KB |
ソースコード
#include <cassert> #include <cmath> #include <algorithm> #include <iostream> #include <iomanip> #include <limits.h> #include <map> #include <queue> #include <set> #include <string.h> #include <vector> using namespace std; typedef long long ll; const int MAX_N = 200010; struct Edge { int v; ll w; Edge(int v = -1, ll w = -1) { this->v = v; this->w = w; } }; vector<Edge> E[MAX_N]; vector<int> V[MAX_N]; ll ans; int N; int dfs1(int u, int parent = -1) { int cnt = 0; for (Edge &e : E[u]) { if (e.v == parent) continue; int v_cnt = dfs1(e.v, u); cnt += v_cnt; V[u].push_back(v_cnt); } return cnt + 1; } void dfs2(int u, int v_cnt = 0, int parent = -1) { int t_cnt = v_cnt; for (int c : V[u]) { t_cnt += c; } int i = 0; for (Edge &e : E[u]) { if (e.v == parent) continue; int cnt = V[u][i]; ans += 2 * e.w * (N - cnt) * cnt; dfs2(e.v, t_cnt - cnt + 1, u); i += 1; } } int main() { cin >> N; int u, v; ll w; for (int i = 0; i < N - 1; ++i) { cin >> u >> v >> w; E[u].push_back(Edge(v, w)); E[v].push_back(Edge(u, w)); } ans = 0; dfs1(1); dfs2(1); cout << ans << endl; return 0; }