题目描述
题目大意
给定 \(n\) 个整数,求每个数和它之前的数的差的绝对值的最小值之和。
题解
暴力是 \(\mathcal{O}(n^2)\) ,本来能卡过的,结果被洛谷改小了时限。
想到对于每个数要快速找到离它最近的数,而平衡树的前驱后继正好满足要求。
可以用vector当平衡树
(multiset也行)
code
785B, 481ms
#include <algorithm>
#include <vector>
#include <cstdio>
#define MAXINT 0x7fffffff
using namespace std;
int Aabs(int x) {
return x >= 0 ? x : -x;
}
int main() {
vector<int> tree;
tree.push_back(-MAXINT);
int n, ans = 0;
scanf("%d", &n);
for (int i = 1; i <= n; ++i) {
int x;
scanf("%d", &x);
if (i == 1) {
ans += x;
} else {
int a = *--lower_bound(tree.begin(), tree.end(), x);
int b = *lower_bound(tree.begin(), tree.end(), x);
if (a == -MAXINT)
ans += Aabs(b - x);
else
ans += min(Aabs(a - x), Aabs(x - b));
}
tree.insert(upper_bound(tree.begin(), tree.end(), x), x);
}
printf("%d\n", ans);
return 0;
}