DTOJ 4655. 「CSP-S 2019」划分

题意

2048 年，第三十届 CSP 认证的考场上，作为选手的小明打开了第一题。这个题的样例有 $n$ 组数据，数据从 $1\sim n$ 编号，$i$ 号数据的规模为 $a_i$。

小明对该题设计出了一个暴力程序，对于一组规模为 $u$ 的数据，该程序的运行时间为 $u^2$ 。然而这个程序运行完一组规模为 $u$ 的数据之后，它将在任何一组规模小于 $u$ 的数据上运行错误。样例中的 $a_i$ 不一定递增，但小明又想在不修改程序的情况下正确运行样例，于是小明决定使用一种非常原始的解决方案：将所有数据划分成若干个数据段，段内数据编号连续，接着将同一段内的数据合并成新数据，其规模等于段内原数据的规模之和，小明将让新数据的规模能够递增。

也就是说，小明需要找到一些分界点 $1 \leq k_1 \lt k_2 \lt \cdots \lt k_p \lt n $，使得

$$\sum _{i=1}^{k_1}{a}_i\le \sum _{i=k_1+1}^{k_2}{a}_i\le \cdots \le \sum _{i=k_p+1}^n{a}_i$$

注意 $p$ 可以为 $0$ 且此时 $k_0=0$，也就是小明可以将所有数据合并在一起运行。

小明希望他的程序在正确运行样例情况下，运行时间也能尽量小，也就是最小化

$$(\sum _{i=1}^{k_1}{a}_i{)}^2+(\sum _{i=k_1+1}^{k_2}{a}_i{)}^2+\cdots +(\sum _{i=k_p+1}^n{a}_i{)}^2$$

小明觉得这个问题非常有趣，并向你请教：给定 $n$ 和 $a_i$ ，请你求出最优划分方案下，小明的程序的最小运行时间。

测试点编号	$n\le$	$a_i\le$	$type=$
$1\sim 3$	10	10	0
$4\sim 6$	50	$10^3$	0
$7\sim 9$	400	$10^4$	0
$10\sim 16$	5000	$10^5$	0
$17\sim 22$	$5\times 10^5$	$10^6$	0
$23\sim 25$	$4\times 10^7$	$10^9$	1

对于$type=0$的所有测试点,保证最后输出的答案 $\le 4\times 10^{18}$

所有测试点满足：
t y p e ∈ { 0 , 1 } ， 2 ≤ n ≤ 4 × 1 0 7 ， 1 ≤ a i ≤ 1 0 9 ， 1 ≤ m ≤ 1 0 5 ， 1 ≤ l i ≤ r i ≤ 1 0 9 ， 0 ≤ x , y , z , b 1 , b 2 < 2 30 type \in \{0,1\}，2 \leq n \leq 4 \times 10^7， 1 \leq a_i \leq 10^9，1 \leq m \leq 10^5，1 \leq l_i \leq r_i \leq 10^9，0 \leq x,y,z,b_1,b_2 \lt 2^{30} type∈{0,1}，2≤n≤4×107，1≤ai​≤109，1≤m≤105，1≤li​≤ri​≤109，0≤x,y,z,b1​,b2​<230

题解

思路比较巧妙的题（显然我是不会的）。

如果DP记录的状态是当前的最小代价，则转移时不可避免地要判大小，这样又要多记一维最后一段的起点来算最后一段的和，而直接贪心又不太行（考场上想到自闭）。

考虑能否把最后一段的和与最小总代价合在一起，结合直观的感受：每段尽量小，可以猜想最后一段最小对应总代价最小，又容易证明其正确性：对前面和后面都是最优，于是DP状态就只有一维，转移时单调队列优化即可。

用两个long long来记答案，要卡一下空间。