結果
問題 | No.2638 Initial fare |
ユーザー | yuyu_5510 |
提出日時 | 2024-02-19 22:05:54 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,285 bytes |
コンパイル時間 | 1,618 ms |
コンパイル使用メモリ | 173,024 KB |
実行使用メモリ | 34,076 KB |
最終ジャッジ日時 | 2024-02-19 22:06:03 |
合計ジャッジ時間 | 6,269 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,676 KB |
testcase_01 | AC | 2 ms
6,676 KB |
testcase_02 | AC | 2 ms
6,676 KB |
testcase_03 | WA | - |
testcase_04 | WA | - |
testcase_05 | WA | - |
testcase_06 | AC | 2 ms
6,676 KB |
testcase_07 | WA | - |
testcase_08 | AC | 130 ms
15,508 KB |
testcase_09 | WA | - |
testcase_10 | AC | 148 ms
16,668 KB |
testcase_11 | AC | 1 ms
6,676 KB |
testcase_12 | WA | - |
testcase_13 | AC | 122 ms
14,916 KB |
testcase_14 | WA | - |
testcase_15 | AC | 139 ms
16,684 KB |
testcase_16 | WA | - |
testcase_17 | WA | - |
testcase_18 | WA | - |
testcase_19 | WA | - |
testcase_20 | WA | - |
testcase_21 | WA | - |
testcase_22 | WA | - |
testcase_23 | WA | - |
testcase_24 | WA | - |
testcase_25 | WA | - |
testcase_26 | WA | - |
testcase_27 | WA | - |
ソースコード
#include<bits/stdc++.h> using namespace std; void dfs(int v, int pre, vector<vector<int>>& g, vector<long long>& ch, long long &ans){ for(auto nv : g[v]){ if(nv == pre) continue; dfs(nv, v, g, ch, ans); ch[v] += ch[nv]; } } void dfs2(int v, int pre, vector<vector<int>>& g, vector<long long>& ch, long long &ans){ for(auto nv : g[v]){ if(nv == pre) continue; dfs2(nv, v, g, ch, ans); } long long sum = 0; for(auto nv : g[v]){ if(nv == pre) continue; sum += ch[nv] - 1; } long long n = (long long)(g.size()); if(pre != -1){ sum += n - ch[v] - 1; } long long minus = 0; for(auto nv : g[v]){ if(nv == pre) continue; minus += (sum - ch[nv] + 1) * (ch[nv] - 1); } if(pre != -1){ minus += (sum - (n - ch[v] - 1)) * (n - ch[v] - 1); } ans -= minus / 2; } int main(){ long long n; cin >> n; vector<vector<int>> g(n); for(int i = 0;i < n-1;i++){ int u, v; cin >> u >> v; u--;v--; g[u].push_back(v); g[v].push_back(u); } long long ans = n*(n-1) / 2; vector<long long> ch(n, 1); dfs(0, -1, g, ch, ans); dfs2(0, -1, g, ch, ans); cout << ans << endl; }