abc173_f Intervals on Tree 容斥,数学思路

题意

给一棵树,定义$f(l,r)$为$[l,r]$区间内的点在树上形成连通块的个数,求${\sum }_{i=1}^n{\sum }_{j=i}^{n}f(i,j)$.

绝杀

abcf题基本上属于思路题,在代码编写上很少有难度(个别超级dp就是另一回事了).
本题先不考虑两点相连减少连通块的情况,则每个点都能形成一个连通块,答案为$\frac{n\times (n+1)\times (n+2)}{6}$.再考虑输入,由于树这个数据结构的特殊性,如果输入两点$u,v(u<v)$之间有一条边,在答案中包含$[u,v]$的区间都会因$u,v$构成连通块而减少一个答案,并且由于树一共只有n-1条边,每个答案一定被且仅被减少一次,因此树上的每一条边减少的答案都可以分开计算.
代码十分简单.这里仅给出main函数中的内容.

int main() {
  int i,n=read(),u,v;
  ll lxy=1ll*n*(n+1)*(n+2)/6;
  for (i=1;i<n;++i) {
    read(u),read(v);
    if (u>v) swap(u,v);
    lxy-=1ll*u*(n-v+1);
  }
  printf("%lld\n",lxy);
}