結果
問題 | No.872 All Tree Path |
ユーザー |
![]() |
提出日時 | 2023-10-15 12:23:31 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 114 ms / 3,000 ms |
コード長 | 1,087 bytes |
コンパイル時間 | 1,101 ms |
コンパイル使用メモリ | 124,008 KB |
最終ジャッジ日時 | 2025-02-17 07:53:32 |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
#include<iostream>#include<vector>#include<algorithm>#include<cstring>#include<cassert>#include<cmath>#include<ctime>#include<iomanip>#include<numeric>#include<stack>#include<queue>#include<map>#include<unordered_map>#include<set>#include<unordered_set>#include<bitset>#include<random>using namespace std;int N,ch[2 << 17];vector<pair<int,int>> G[2 << 17];long long dp[2 << 17];void dfs0(int u,int p,long long W){dp[u] = W;ch[u] = 1;for(auto [v,w] : G[u]) if(v != p){dfs0(v,u,W+w);dp[u] += dp[v];ch[u] += ch[v];}}long long ans;void dfs1(int u,int p,int W){if(p >= 0) dp[u] += (dp[p]-dp[u]+(long long)W*(N-2*ch[u]));ans += dp[u];for(auto [v,w]:G[u]) if(v != p) dfs1(v,u,w);}void solve(){cin >> N;for(int i = 1;i < N;i++){int u,v,w;cin >> u >> v >> w;u--; v--;G[u].push_back(make_pair(v,w));G[v].push_back(make_pair(u,w));}dfs0(0,-1,0);ans = 0;dfs1(0,-1,0);cout << ans << endl;}int main(){ios::sync_with_stdio(false);cin.tie(nullptr);//tanosimuint tt = 1;//cin >> tt;while(tt--) solve();}