結果
問題 | No.1103 Directed Length Sum |
ユーザー |
![]() |
提出日時 | 2020-07-03 21:32:29 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 650 ms / 3,000 ms |
コード長 | 930 bytes |
コンパイル時間 | 1,551 ms |
コンパイル使用メモリ | 175,064 KB |
実行使用メモリ | 116,480 KB |
最終ジャッジ日時 | 2024-09-16 22:58:54 |
合計ジャッジ時間 | 7,241 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 |
コンパイルメッセージ
main.cpp: In function 'int main()': main.cpp:45:6: warning: 'r' may be used uninitialized [-Wmaybe-uninitialized] 45 | dfs(sz, c, r); | ~~~^~~~~~~~~~ main.cpp:26:7: note: 'r' was declared here 26 | int r; | ^
ソースコード
#include <bits/stdc++.h> using namespace std; long long MOD = 1000000007; int dfs(vector<int> &sz, vector<vector<int>> &c, int v){ sz[v] = 1; for (int w : c[v]){ sz[v] += dfs(sz, c, w); } return sz[v]; } int main(){ ios::sync_with_stdio(false); cin.tie(nullptr); int N; cin >> N; vector<int> p(N, -1); vector<vector<int>> c(N); for (int i = 0; i < N - 1; i++){ int A, B; cin >> A >> B; A--; B--; p[B] = A; c[A].push_back(B); } int r; for (int i = 0; i < N; i++){ if (p[i] == -1){ r = i; } } vector<int> d(N); d[r] = 0; queue<int> Q; Q.push(r); while (!Q.empty()){ int v = Q.front(); Q.pop(); for (int w : c[v]){ d[w] = d[v] + 1; Q.push(w); } } vector<int> sz(N, 0); dfs(sz, c, r); long long ans = 0; for (int i = 0; i < N; i++){ ans += (long long) d[i] * sz[i]; ans %= MOD; } cout << ans << endl; }