結果

問題 No.2638 Initial fare
コンテスト
ユーザー tobbie
提出日時 2024-05-06 18:23:12
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 201 ms / 2,000 ms
コード長 574 bytes
コンパイル時間 1,725 ms
コンパイル使用メモリ 170,376 KB
実行使用メモリ 14,880 KB
最終ジャッジ日時 2024-11-29 03:12:43
合計ジャッジ時間 7,609 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
using ll = long long int;

int main() {
  int N;
  cin >> N;
  vector<vector<int>> g(N);
  rep(i, N-1) {
    int u, v;
    cin >> u >> v;
    u--; v--;
    g[u].push_back(v);
    g[v].push_back(u);
  }
  ll ans = 0;
  ans += N - 1;
  rep(i, N)
    ans += ((ll)g[i].size() * ((ll)g[i].size() - 1)) / 2;
  ll ans2 = 0;
  rep(i, N) {
    for (int j : g[i]) {
      ans2 += ((ll)g[i].size() - 1) * ((ll) g[j].size() - 1);
    }
  }
  cout << ans + ans2/2 << endl;
  return 0;
}
0