【洛谷】P2234 营业额统计

题目地址：

https://www.luogu.com.cn/problem/P2234

题目描述：
Tiger最近被公司升任为营业部经理，他上任后接受公司交给的第一项任务便是统计并分析公司成立以来的营业情况。Tiger拿出了公司的账本，账本上记录了公司成立以来每天的营业额。分析营业情况是一项相当复杂的工作。由于节假日，大减价或者是其他情况的时候，营业额会出现一定的波动，当然一定的波动是能够接受的，但是在某些时候营业额突变得很高或是很低，这就证明公司此时的经营状况出现了问题。经济管理学上定义了一种最小波动值来衡量这种情况：当最小波动值越大时，就说明营业情况越不稳定。而分析整个公司的从成立到现在营业情况是否稳定，只需要把每一天的最小波动值加起来就可以了。你的任务就是编写一个程序帮助Tiger来计算这一个值。我们定义，设第$i$天营业额为$x_i$，则第$i$天的最小波动值为 min ⁡ j < i { ∣ x i − x j ∣ } \min_{j<i}\{|x_i-x_j|\} minj<i​{∣xi​−xj​∣}。特别地，第一天的最小波动值为第一天的营业额。

输入格式：
第一行为正整数$n$（$n\le 32767$），表示该公司从成立一直到现在的天数，接下来的$n$行每行有一个整数$a_i$（$|a_i|\le 10^6$)，表示第$i$天公司的营业额，可能存在负数。

输出格式：
输出一个正整数，即每一天最小波动值的和，保证结果小于$2^{31}$。

可以用平衡树set来做，代码如下：

#include <iostream>
#include <set>
using namespace std;

int n;
set<int> st;
int res;

int main() {
  scanf("%d", &n);
  while (n--) {
    int x;
    scanf("%d", &x);
    int diff = 1e9;
    auto it1 = st.lower_bound(x);
    if (it1 != st.end()) diff = min(diff, abs(*it1 - x));
    if (it1 != st.begin()) {
      auto it2 = prev(it1);
      diff = min(diff, abs(*it2 - x));
    }
    res += diff == (int)1e9 ? x : diff;
    st.insert(x);
  }

  printf("%d\n", res);
}

时间复杂度$O(n\log n)$，空间$O(n)$。