J. 上学

原题链接

题解

看清楚题目，是三个人都坐在同一辆校车！！

code

#include<bits/stdc++.h>
#define ll long long
using namespace std;
vector<ll> G[200005];
ll depth[200005]={0};
void dfs(ll now,ll fa)
{for(auto next:G[now]){if(next==fa) continue;depth[next]=depth[now]+1;dfs(next,now);}
}ll C(ll a,ll b)
{ll sum=1;for(ll i=0;i<b;i++){sum=sum*(a-i);}for(ll i=2;i<=b;i++) sum=sum/i;return sum;
}
int main()
{ll n;cin>>n;for(ll i=1;i<n;i++){ll x,y;cin>>x>>y;G[x].push_back(y);G[y].push_back(x);}dfs(1,0);ll cnt=0;for(ll i=1;i<=n;i++){if(depth[i]>=2) cnt+=C(depth[i],2);}cout<<C(n,3)-cnt;return 0;
}