#ABC359F. 树的度数优化

树的度数优化

树的度数优化

题目描述

给定整数序列 A=(A1,…,AN)A=(A_1,\ldots,A_N)。对于一棵有 NN 个顶点的树 TT,定义 f(T)f(T) 如下:

设 did_i 为 TT 中顶点 ii 的度数,则

f(T)=∑i=1Ndi2Aif(T)=\sum_{i=1}^N {d_i}^2 A_i

求 f(T)f(T) 的最小可能值。

题目保证答案小于 2632^{63}。

输入格式

输入按以下格式从标准输入给出:

NN
A1A_1 A2A_2 …\ldots ANA_N

输出格式

输出答案。

样例

4
3 2 5 2
24

考虑一棵树 TT,其边为:顶点 11 与 22 相连,顶点 22 与 44 相连,顶点 44 与 33 相连。

此时,$f(T)=1^2\times3+2^2\times2+1^2\times5+2^2\times2=24$。可以证明这就是 f(T)f(T) 的最小值。

3
4 3 2
15
7
10 5 10 2 10 13 15
128

数据范围

  • 2≤N≤2×1052 \le N \le 2\times10^5
  • 1≤Ai≤1091 \le A_i \le 10^9
  • 输入均为整数。
难度 提高+/省选
通过率 —
尝试 0
通过 0
ID
3338
类型
传统题
Time Limit
2000ms
Memory Limit
1024MiB
上传者
标签