【ACWing】531. 铺设道路（配数学证明）

题目地址：

https://www.acwing.com/problem/content/description/533/

春春是一名道路工程师，负责铺设一条长度为$n$的道路。铺设道路的主要工作是填平下陷的地表。整段道路可以看作是$n$块首尾相连的区域，一开始，第$i$块区域下陷的深度为$d_i$。春春每天可以选择一段连续区间$[L,R]$，填充这段区间中的每块区域，让其下陷深度减少$1$。在选择区间时，需要保证，区间内的每块区域在填充前下陷深度均不为$0$。春春希望你能帮他设计一种方案，可以在最短的时间内将整段道路的下陷深度都变为$0$。

输入格式：
输入文件包含两行，第一行包含一个整数$n$，表示道路的长度。 
第二行包含$n$个整数，相邻两数间用一个空格隔开，第$i$个整数为$d_i$。

输出格式：
输出文件仅包含一个整数，即最少需要多少天才能完成任务。

数据范围：
$1\le n\le 10^5$
$0\le d_i\le 10000$

问题可以转述为，给定一个长$n$的非负整数数组$A$，每次允许选取一段，将这一段里的每个数都减$1$，问至少减多少次就能将$A$变为零数组。可以证明，答案为 ∑ i = 1 n max ⁡ { A i − A i − 1 , 0 } \sum_{i=1}^n \max\{A_i-A_{i-1},0\} ∑i=1n​max{Ai​−Ai−1​,0}，其中$A$下标从$1$开始，定义$A_0=0$。

算法正确性证明：
可以用数学归纳法。

先证明做 ∑ i = 1 n max ⁡ { A i − A i − 1 , 0 } \sum_{i=1}^n \max\{A_i-A_{i-1},0\} ∑i=1n​max{Ai​−Ai−1​,0}次操作确实可以达到目的。考虑从$A_1$开始的向右的最长非严格上升的一段，比如$A_1\le A_2\le ...\le A_k>A_{k+1}$，由于我们要将每个数都减成$0$，那么为了将$A_k$减为$0$，我们至少需要$A_k$次操作。无论什么样的方案，操作的进行次序是可以交换的（因为每次操作的实质是将向量$A$加上另一个向量，向量的加法有交换律），所以我们把含将$A_k$减$1$的操作换到最前面做，并且让这些操作将$A_{1\sim k-1}$也变为$0$，右边不动（这显然是可以做到的），做完这$A_k$次操作之后，由归纳假设，剩余部分需要 ∑ i = k + 1 n max ⁡ { A i − A i − 1 , 0 } \sum_{i=k+1}^n \max\{A_i-A_{i-1},0\} ∑i=k+1n​max{Ai​−Ai−1​,0}次就可以达到目的，总次数等于 A k + ∑ i = k + 1 n max ⁡ { A i − A i − 1 , 0 } = ∑ i = 1 n max ⁡ { A i − A i − 1 , 0 } A_k+\sum_{i=k+1}^n \max\{A_i-A_{i-1},0\}=\sum_{i=1}^n \max\{A_i-A_{i-1},0\} Ak​+∑i=k+1n​max{Ai​−Ai−1​,0}=∑i=1n​max{Ai​−Ai−1​,0}，成立。

再证明 ∑ i = 1 n max ⁡ { A i − A i − 1 , 0 } \sum_{i=1}^n \max\{A_i-A_{i-1},0\} ∑i=1n​max{Ai​−Ai−1​,0}次是最少的次数。考虑任意一种方案，将所有对$A_n$的操作分解为两个操作，一个是对前面的操作，另一个是单独仅仅对$A_n$的操作。那么由归纳假设，如果忽略$A_n$，要将$A_{1\sim n-1}$变为$0$至少需要的次数是 ∑ i = 1 n − 1 max ⁡ { A i − A i − 1 , 0 } \sum_{i=1}^{n-1} \max\{A_i-A_{i-1},0\} ∑i=1n−1​max{Ai​−Ai−1​,0}。单独对$A_n$的操作次数显然就是$A_n$，这些操作有$A_{n-1}$个可以匀给含对$A_{n-1}$的操作里一起做，所以总次数一定不少于 ∑ i = 1 n max ⁡ { A i − A i − 1 , 0 } \sum_{i=1}^n \max\{A_i-A_{i-1},0\} ∑i=1n​max{Ai​−Ai−1​,0}。

综上，算法成立。代码如下：

#include<iostream>
using namespace std;

const int N = 1e5 + 10;
int n;
int a[N], res;

int main() {
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
    for (int i = 1; i <= n; i++) res += max(a[i] - a[i - 1], 0);
    printf("%d\n", res);
    return 0;
}

时空复杂度$O(n)$。