結果
問題 | No.2638 Initial fare |
ユーザー | ripity |
提出日時 | 2024-02-19 21:48:39 |
言語 | C++17(gcc12) (gcc 12.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 200 ms / 2,000 ms |
コード長 | 617 bytes |
コンパイル時間 | 2,166 ms |
コンパイル使用メモリ | 205,104 KB |
実行使用メモリ | 16,460 KB |
最終ジャッジ日時 | 2024-09-29 01:45:26 |
合計ジャッジ時間 | 6,542 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 25 |
ソースコード
#include <bits/stdc++.h> using namespace std; int main() { int N; cin >> N; vector<int> A(N), B(N); vector<vector<int>> G(N); for( int i = 0; i < N-1; i++ ) { cin >> A[i] >> B[i]; A[i]--, B[i]--; G[A[i]].push_back(i); G[B[i]].push_back(i); } long long ans = N-1; for( int i = 0; i < N; i++ ) { long long deg = G[i].size(); ans += deg*(deg-1)/2; } for( int i = 0; i < N-1; i++ ) { long long dega = G[A[i]].size(); long long degb = G[B[i]].size(); ans += (dega-1)*(degb-1); } cout << ans << endl; }