結果

問題 No.2638 Initial fare
コンテスト
ユーザー ゆにぽけ
提出日時 2024-02-19 21:26:30
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,187 bytes
コンパイル時間 1,304 ms
コンパイル使用メモリ 127,712 KB
最終ジャッジ日時 2025-02-19 16:35:52
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 4 TLE * 1 -- * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <array>
#include <iterator>
#include <string>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <cassert>
#include <cmath>
#include <ctime>
#include <iomanip>
#include <numeric>
#include <stack>
#include <queue>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <bitset>
#include <random>
#include <utility>
#include <functional>
using namespace std;
void Main()
{
	int N;
	cin >> N;
	vector<int> U(N),V(N);
	vector<vector<int>> G(N);
	for(int i = 1;i < N;i++)
	{
		cin >> U[i] >> V[i];
		U[i]--; V[i]--;
		G[U[i]].push_back(V[i]);
		G[V[i]].push_back(U[i]);
	}
	long long ans = 0;
	for(int i = 1;i < N;i++)
	{
		int a = 0,b = 0;
		int u = U[i],v = V[i];
		for(int nu : G[u])
		{
			if(nu != v)
			{
				a++;
			}
		}
		for(int nv : G[v])
		{
			if(nv != u)
			{
				b++;
			}
		}
		ans += (long long) a * b;
		ans++;
	}
	for(int i = 0;i < N;i++)
	{
		int deg = (int)G[i].size();
		ans += (long long)deg * (deg - 1) / 2;
	}
	cout << ans << "\n";
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int tt = 1;
	/* cin >> tt; */
	while(tt--) Main();
}
0