結果

問題 No.2638 Initial fare
ユーザー daiotadaiota
提出日時 2024-02-19 22:19:39
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 683 bytes
コンパイル時間 1,721 ms
コンパイル使用メモリ 178,532 KB
実行使用メモリ 40,004 KB
最終ジャッジ日時 2024-09-29 02:14:35
合計ジャッジ時間 7,931 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other WA * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef pair<int,int> P;
#define REP(i,n) for(int i=0;i<int(n);i++)

vector<ll> G[200010];
ll dep[200010];
ll p[200010];

void dfs(ll v,ll b,ll d){
	p[v]=b;
	dep[v]=d;
	for(ll i=0;i<(ll)G[v].size();i++) if(G[v][i]!=b) dfs(G[v][i],v,d+1);

}

int main(){
	cin.tie(nullptr);  ios_base::sync_with_stdio(false);
   ll i,j;

	ll N;
	cin >> N;

	for(i=1;i<N;i++){
		ll a,b;
		cin >> a >> b;
		G[a].push_back(b);
		G[b].push_back(a);
	}

	dfs(1,-1,0);

	map<ll,ll> m;
	for(i=1;i<=N;i++){
		m[dep[i]]++;
	}

	ll ans=0;
	for(i=0;i<N;i++){
		ans+=m[i]*(m[i+1]+m[i+2]+m[i+3]);
	}


	cout << ans << endl;






	return 0;

}
0