题意
给一棵树,定义 f ( l , r ) f(l,r) f(l,r)为 [ l , r ] [l,r] [l,r]区间内的点在树上形成连通块的个数,求 ∑ i = 1 n ∑ j = i n f ( i , j ) \sum^n_{i=1}\sum^n_{j=i}f(i,j) ∑i=1n∑j=inf(i,j).
绝杀
abcf题基本上属于思路题,在代码编写上很少有难度(个别超级dp就是另一回事了).
本题先不考虑两点相连减少连通块的情况,则每个点都能形成一个连通块,答案为
n
×
(
n
+
1
)
×
(
n
+
2
)
6
\frac{n \times (n+1) \times (n+2)}{6}
6n×(n+1)×(n+2).再考虑输入,由于树这个数据结构的特殊性,如果输入两点
u
,
v
(
u
<
v
)
u,v(u<v)
u,v(u<v)之间有一条边,在答案中包含
[
u
,
v
]
[u,v]
[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);
}