結果
問題 | No.1507 Road Blocked |
ユーザー |
![]() |
提出日時 | 2021-05-14 22:21:46 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 355 ms / 2,000 ms |
コード長 | 869 bytes |
コンパイル時間 | 6,348 ms |
コンパイル使用メモリ | 198,856 KB |
最終ジャッジ日時 | 2025-01-21 11:27:34 |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/modint> using namespace std; using namespace atcoder; using mint = modint998244353; using graph = vector<vector<int>>; int n; mint dfs(const graph &to, vector<int> &sz, int v, int p) { mint res = 0; for (auto &&c : to.at(v)) { if (c == p) { continue; } res += dfs(to, sz, c, v); sz.at(v) += sz.at(c); res += mint(n) * (n - 1) / 2 - mint(sz.at(c)) * (n - sz.at(c)); } sz.at(v)++; return res; } int main() { cin >> n; graph to(n); for (int i = 0; i < n - 1; i++) { int u, v; cin >> u >> v; u--; v--; to.at(u).push_back(v); to.at(v).push_back(u); } vector<int> sz(n); cout << (dfs(to, sz, 0, -1) / (mint(n) * (n - 1) * (n - 1) / 2)).val() << endl; return 0; }