結果
問題 | No.1507 Road Blocked |
ユーザー |
![]() |
提出日時 | 2021-05-15 19:08:26 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 45 ms / 2,000 ms |
コード長 | 1,550 bytes |
コンパイル時間 | 1,004 ms |
コンパイル使用メモリ | 96,652 KB |
実行使用メモリ | 10,368 KB |
最終ジャッジ日時 | 2024-10-03 06:19:52 |
合計ジャッジ時間 | 4,193 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
/* -*- coding: utf-8 -*- * * 1507.cc: No.1507 Road Blocked - yukicoder */ #include<cstdio> #include<cstdlib> #include<cstring> #include<cmath> #include<iostream> #include<string> #include<vector> #include<map> #include<set> #include<stack> #include<list> #include<queue> #include<deque> #include<algorithm> #include<numeric> #include<utility> #include<complex> #include<functional> using namespace std; /* constant */ const int MAX_N = 100000; const int MOD = 998244353; /* typedef */ typedef long long ll; typedef vector<int> vi; /* global variables */ vi nbrs[MAX_N]; int ps[MAX_N], cis[MAX_N], ss[MAX_N]; /* subroutines */ int powmod(int a, int n) { // a^n % MOD int pm = 1; while (n > 0) { if (n & 1) pm = (ll)pm * a % MOD; a = (ll)a * a % MOD; n >>= 1; } return pm; } inline int p2(int n) { return (ll)n * (n - 1) % MOD; } /* main */ int main() { int n; scanf("%d", &n); for (int i = 1; i < n; i++) { int u, v; scanf("%d%d", &u, &v); u--, v--; nbrs[u].push_back(v); nbrs[v].push_back(u); } int sum = 0; ps[0] = -1; for (int u = 0; u >= 0;) { vi &nbru = nbrs[u]; int up = ps[u]; if (cis[u] < nbru.size()) { int v = nbru[cis[u]++]; if (v != up) { ps[v] = u; u = v; } } else { ss[u]++; if (up >= 0) { ss[up] += ss[u]; sum = (sum + (p2(ss[u]) + p2(n - ss[u])) % MOD) % MOD; } u = up; } } int num = (ll)p2(n) * (n - 1) % MOD; printf("%lld\n", (ll)sum * powmod(num, MOD - 2) % MOD); return 0; }