难度:5
动态查找前驱后继,可以用链表的离线做法,这个是用平衡树的在线做法,这道题的模型就是按照输入顺序,找之前输入的一个数使得和现在数的差的绝对值最小,然后输出和,用平衡树可以很方便的找到前驱后继,
#include <bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define all(x) (x).begin(), (x).end()
using namespace std;
typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> pa;
const int maxn = 33005;
int main() {
int n;
cin >> n;
ll ans = 0;
multiset<int> st;
for (int i = 1; i <= n; i++) {
int x;
cin >> x;
if (i == 1) { ans += x; st.insert(x); continue; }
set<int>::iterator it = st.lower_bound(x);
if (it == st.end()) { it--; ans += abs(*it - x); }
else if (it == st.begin()) ans += abs(*it - x);
else {
set<int>::iterator it2 = it; it2--;
ans += min(abs(*it - x), abs(*it2 - x));
}
st.insert(x);
}
cout << ans;
return 0;
}