Comet OJ - Contest #6 一道树题

https://www.cometoj.com/contest/48/problem/C?problem_id=2279 

 总结：连通块数=点数-边数

要求全部区间点数-边数，可以先将这些区间点数求出来，再减这些区间的边数；点数的求法就是看该点对哪些区间有贡献，容斥用全部区间数减掉不包含该点的区间数，这样就可以得到全部区间的全部点数（一个点可以贡献多个区间）；

边数：就是区间长度总和，sum(k*(n-k));

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <queue>
#include <map>
#include <set>
#include <stack>
using namespace std;

#define sfi(x) scanf("%d",&x)
#define sfc(x) scanf("%c",x)
#define sfl(x) scanf("%lld",&x)
#define sfs(x) scanf("%s",x)

#define rint register int
#define pb push_back
#define fl() printf("flag!\n")
#define INF 0x3f3f3f3f
#define ll long long
#define mem(x,y) memset(x,y,sizeof(x))
#define FAST_IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)

const int maxn=2e5+9;


ll cal(ll x)
{
    return x*(x+1)/2;
}

vector<int>g[maxn];

int main()
{
    //FAST_IO;
    //freopen("input.txt","r",stdin);

    ll n;
    cin>>n;
    for(int i=1;i<=n-1;i++)
    {
        int u,v;
        cin>>u>>v;
        g[u].pb(i);
        g[v].pb(i);
    }
    ll ans=0;//合法区间点总和
    ll sum=(n-1)*n/2;//区间总数
    for(int i=1;i<=n;i++)
    {
        int pre=0;
        ans+=sum;//加上区间总数然后减掉不包含该点的区间数(就是两点间的区间数)
        for(int j=0;j<g[i].size();j++)
        {
            ans-=cal(g[i][j]-pre-1);
            pre=g[i][j];
        }
        ans-=cal(n-1-pre);
    }

    for(ll i=1;i<=n-1;i++)
    {
        ans-=i*(n-i);//减掉边总和
    }

    cout<<ans<<endl;

    return 0;
}