結果
問題 | No.1103 Directed Length Sum |
ユーザー |
![]() |
提出日時 | 2020-07-03 22:10:16 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 658 ms / 3,000 ms |
コード長 | 1,741 bytes |
コンパイル時間 | 1,621 ms |
コンパイル使用メモリ | 169,488 KB |
実行使用メモリ | 152,960 KB |
最終ジャッジ日時 | 2024-09-17 02:29:29 |
合計ジャッジ時間 | 8,655 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 |
ソースコード
#include<bits/stdc++.h> using namespace std; #define int long long #define ii pair <int, int> #define app push_back #define all(a) a.begin(), a.end() #define bp __builtin_popcountll #define ll long long #define mp make_pair #define f first #define s second #define Time (double)clock()/CLOCKS_PER_SEC #define debug(x) std::cout << #x << ": " << x << '\n'; const int N = 1e6+7, MOD = 1000 * 1000 * 1000 + 7; int mod(int n) { n %= MOD; if (n < 0) return n + MOD; else return n; } int fp(int a, int p) { int ans = 1, c = a; for (int i = 0; (1ll << i) <= p; ++i) { if ((p >> i) & 1) ans = mod(ans * c); c = mod(c * c); } return ans; } int dv(int a, int b) { return mod(a * fp(b, MOD - 2)); } int cnt[N], dist[N]; bool used[N]; vector <int> g[N]; void dfs(int u) { used[u] = 1; for (int v : g[u]) { if (!used[v]) { dfs(v); } cnt[u] += cnt[v]; dist[u] += dist[v]; cnt[u] = mod(cnt[u]); dist[u] = mod(dist[u]); } dist[u] += cnt[u]; ++cnt[u]; cnt[u] = mod(cnt[u]); dist[u] = mod(dist[u]); } signed main() { #ifdef HOME freopen("input.txt", "r", stdin); #else #define endl '\n' ios_base::sync_with_stdio(0); cin.tie(0); #endif int n; cin >> n; for (int i = 0; i < n - 1; ++i) { int u, v; cin >> u >> v; g[u].app(v); } int ans = 0; for (int i = 1; i <= n; ++i) { if (!used[i]) { dfs(i); } } for (int i = 1; i <= n; ++i) ans += dist[i]; ans = mod(ans); cout << ans << endl; }