結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
![]() |
提出日時 | 2021-03-05 21:31:45 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 105 ms / 2,000 ms |
コード長 | 710 bytes |
コンパイル時間 | 829 ms |
コンパイル使用メモリ | 86,376 KB |
実行使用メモリ | 21,248 KB |
最終ジャッジ日時 | 2024-10-07 00:39:25 |
合計ジャッジ時間 | 3,776 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include <iostream> #include <set> #include <map> #include <cmath> #include <queue> #include <vector> #include <string> #include <algorithm> #include <functional> using namespace std; long long N, A[1 << 18], B[1 << 18]; long long dp[1 << 18]; vector<int> X[1 << 18]; void dfs(int pos, int pre) { dp[pos] = 1; for (int i : X[pos]) { if (i == pre) continue; dfs(i, pos); dp[pos] += dp[i]; } } int main() { cin >> N; for (int i = 1; i <= N - 1; i++) { cin >> A[i] >> B[i]; X[A[i]].push_back(B[i]); X[B[i]].push_back(A[i]); } dfs(1, -1); long long Answer = 0; for (int i = 2; i <= N; i++) { Answer += 2LL * dp[i] * (N - dp[i]); } Answer += N * N; cout << Answer << endl; return 0; }