結果

問題 No.2638 Initial fare
コンテスト
ユーザー cho435
提出日時 2024-02-20 23:51:06
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 421 ms / 2,000 ms
コード長 554 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 4,548 ms
コンパイル使用メモリ 250,168 KB
最終ジャッジ日時 2025-02-19 18:12:01
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using ll = long long;
#define rep(i,n) for(int i=0;i<(int)(n);i++)
using mint = atcoder::modint998244353;

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