J. 上學

纯粹的發表於2024-05-14

原題連結

題解

看清楚題目,是三個人都坐在同一輛校車!!

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;
}

相關文章