[P3453] [POI2007] DRZ - Trees
毫无技术含量的题目。
首先,特判 和 的答案,还有相邻的贡献。
然后记 ,记 ,记 ,考虑 和 所作贡献():
(1),
(2),
(3),
(4),
(5),
(6),
(7),
(8),
(9),
把 看作修改,把 看作询问。
对应的修改看成 到 的线段(限制为 )。
对应的询问看成 到 的线段(限制为 )。
两个线段有交即产生贡献,直接对 轴扫描线即可。
为了满足 ,所以首先离散化的时候不应该去重,然后询问的时候特判 和 两个纵坐标不计算贡献。 计算贡献无所谓,因为其贡献为 。
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int MAXN = 5e4 + 10, INF = numeric_limits<int>::max();
int n, a[MAXN], s[MAXN], id[MAXN], rev[MAXN], sz;
pair<int, int> bd[MAXN];
ll ans[MAXN], sans;
vector<pair<int, int>> op[MAXN];
inline void chsmin(ll &x, const ll &y) { y < x && (x = y); }
ll calc(int x, int y) {
ll ret = 0;
if (x != 1) ret += abs(a[y] - a[x - 1]) - abs(a[x] - a[x - 1]);
if (y != n) ret += abs(a[x] - a[y + 1]) - abs(a[y] - a[y + 1]);
if (x + 1 != y) {
ret += abs(a[x] - a[y - 1]) - abs(a[y] - a[y - 1]) +
abs(a[y] - a[x + 1]) - abs(a[x] - a[x + 1]);
}
return ret;
}
namespace sgt {
int mn[MAXN * 4];
inline int ls(int p) { return p << 1; }
inline int rs(int p) { return p << 1 | 1; }
inline void push_up(int p) { mn[p] = min(mn[ls(p)], mn[rs(p)]); }
void build(int p, int l, int r) {
mn[p] = INF;
if (l == r) return;
int mid = (l + r) >> 1;
build(ls(p), l, mid);
build(rs(p), mid + 1, r);
}
void update(int p, int l, int r, int x, int v) {
if (l == r) {
mn[p] = v;
return;
}
int mid = (l + r) >> 1;
if (x <= mid) {
update(ls(p), l, mid, x, v);
} else {
update(rs(p), mid + 1, r, x, v);
}
push_up(p);
}
int query(int p, int l, int r, int ql, int qr) {
if (ql > qr) return INF;
if (ql <= l && r <= qr) return mn[p];
int mid = (l + r) >> 1;
if (qr <= mid) {
return query(ls(p), l, mid, ql, qr);
} else if (ql > mid) {
return query(rs(p), mid + 1, r, ql, qr);
} else {
return min(query(ls(p), l, mid, ql, qr),
query(rs(p), mid + 1, r, ql, qr));
}
}
} // namespace sgt
int query(int p, int l, int r) {
int x = bd[p].first, y = bd[p].second;
if (r == y) --r;
if (r == x) --r;
if (l == x) ++l;
if (l == y) ++l;
if (l > r) return INF;
if ((x < l && r < y) || r < x || l > y) return sgt::query(1, 1, n, l, r);
if (l < x && r > y) {
return min({sgt::query(1, 1, n, l, x - 1),
sgt::query(1, 1, n, x + 1, y - 1),
sgt::query(1, 1, n, y + 1, r)});
}
if (l < x) {
return min(sgt::query(1, 1, n, l, x - 1),
sgt::query(1, 1, n, x + 1, r));
}
return min(sgt::query(1, 1, n, l, y - 1), sgt::query(1, 1, n, y + 1, r));
}
void solve1() {
for (int i = 1; i <= n; ++i) op[i].clear();
for (int i = 2; i < n; ++i) {
op[1].emplace_back(id[i], a[i - 1] + a[i + 1] - a[i] * 2 - s[i]);
op[bd[i].first].emplace_back(id[i], INF);
}
sgt::build(1, 1, n);
for (int i = 1; i <= n; ++i) {
for (auto &j : op[i]) sgt::update(1, 1, n, j.first, j.second);
int p = rev[i];
if (p != 1 && p != n) {
chsmin(ans[p], (ll)query(p, 1, bd[p].first - 1) + a[p - 1] +
a[p + 1] - a[p] * 2 - s[p]);
}
}
}
void solve2() {
for (int i = 1; i <= n; ++i) op[i].clear();
for (int i = 2; i < n; ++i) {
op[bd[i].first].emplace_back(
id[i], abs(a[i - 1] - a[i + 1]) - a[i] * 2 - s[i]);
op[bd[i].second].emplace_back(id[i], INF);
}
sgt::build(1, 1, n);
for (int i = 1; i <= n; ++i) {
for (auto &j : op[i]) sgt::update(1, 1, n, j.first, j.second);
int p = rev[i];
if (p != 1 && p != n) {
chsmin(ans[p], (ll)query(p, 1, bd[p].first - 1) + a[p - 1] +
a[p + 1] - s[p]);
}
}
}
void solve3() {
for (int i = 1; i <= n; ++i) op[i].clear();
for (int i = 2; i < n; ++i) {
op[bd[i].second].emplace_back(id[i],
-a[i - 1] - a[i + 1] - a[i] * 2 - s[i]);
}
sgt::build(1, 1, n);
for (int i = 1; i <= n; ++i) {
for (auto &j : op[i]) sgt::update(1, 1, n, j.first, j.second);
int p = rev[i];
if (p != 1 && p != n) {
chsmin(ans[p], (ll)query(p, 1, bd[p].first - 1) + a[p - 1] +
a[p + 1] + a[p] * 2 - s[p]);
}
}
}
void solve4() {
for (int i = 1; i <= n; ++i) op[i].clear();
for (int i = 2; i < n; ++i) {
op[1].emplace_back(id[i], a[i - 1] + a[i + 1] - s[i]);
op[bd[i].first].emplace_back(id[i], INF);
}
sgt::build(1, 1, n);
for (int i = 1; i <= n; ++i) {
for (auto &j : op[i]) sgt::update(1, 1, n, j.first, j.second);
int p = rev[i];
if (p != 1 && p != n) {
chsmin(ans[p], (ll)query(p, bd[p].first, bd[p].second - 1) +
abs(a[p - 1] - a[p + 1]) - a[p] * 2 - s[p]);
}
}
}
void solve5() {
for (int i = 1; i <= n; ++i) op[i].clear();
for (int i = 2; i < n; ++i) {
op[bd[i].first].emplace_back(id[i], abs(a[i - 1] - a[i + 1]) - s[i]);
op[bd[i].second].emplace_back(id[i], INF);
}
sgt::build(1, 1, n);
for (int i = 1; i <= n; ++i) {
for (auto &j : op[i]) sgt::update(1, 1, n, j.first, j.second);
int p = rev[i];
if (p != 1 && p != n) {
chsmin(ans[p], (ll)query(p, bd[p].first, bd[p].second - 1) +
abs(a[p - 1] - a[p + 1]) - s[p]);
}
}
}
void solve6() {
for (int i = 1; i <= n; ++i) op[i].clear();
for (int i = 2; i < n; ++i)
op[bd[i].second].emplace_back(id[i], -a[i - 1] - a[i + 1] - s[i]);
sgt::build(1, 1, n);
for (int i = 1; i <= n; ++i) {
for (auto &j : op[i]) sgt::update(1, 1, n, j.first, j.second);
int p = rev[i];
if (p != 1 && p != n) {
chsmin(ans[p], (ll)query(p, bd[p].first, bd[p].second - 1) +
abs(a[p - 1] - a[p + 1]) + a[p] * 2 - s[p]);
}
}
}
void solve7() {
for (int i = 1; i <= n; ++i) op[i].clear();
for (int i = 2; i < n; ++i) {
op[1].emplace_back(id[i], a[i - 1] + a[i + 1] + a[i] * 2 - s[i]);
op[bd[i].first].emplace_back(id[i], INF);
}
sgt::build(1, 1, n);
for (int i = 1; i <= n; ++i) {
for (auto &j : op[i]) sgt::update(1, 1, n, j.first, j.second);
int p = rev[i];
if (p != 1 && p != n) {
chsmin(ans[p], (ll)query(p, bd[p].second, n) - a[p - 1] - a[p + 1] -
a[p] * 2 - s[p]);
}
}
}
void solve8() {
for (int i = 1; i <= n; ++i) op[i].clear();
for (int i = 2; i < n; ++i) {
op[bd[i].first].emplace_back(
id[i], abs(a[i - 1] - a[i + 1]) + a[i] * 2 - s[i]);
op[bd[i].second].emplace_back(id[i], INF);
}
sgt::build(1, 1, n);
for (int i = 1; i <= n; ++i) {
for (auto &j : op[i]) sgt::update(1, 1, n, j.first, j.second);
int p = rev[i];
if (p != 1 && p != n) {
chsmin(ans[p],
(ll)query(p, bd[p].second, n) - a[p - 1] - a[p + 1] - s[p]);
}
}
}
void solve9() {
for (int i = 1; i <= n; ++i) op[i].clear();
for (int i = 2; i < n; ++i) {
op[bd[i].second].emplace_back(id[i],
-a[i - 1] - a[i + 1] + a[i] * 2 - s[i]);
}
sgt::build(1, 1, n);
for (int i = 1; i <= n; ++i) {
for (auto &j : op[i]) sgt::update(1, 1, n, j.first, j.second);
int p = rev[i];
if (p != 1 && p != n) {
chsmin(ans[p], (ll)query(p, bd[p].second, n) - a[p - 1] - a[p + 1] +
a[p] * 2 - s[p]);
}
}
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
cin >> n;
for (int i = 1; i <= n; ++i) cin >> a[i];
iota(rev + 1, rev + n + 1, 1);
sort(rev + 1, rev + n + 1, [](int x, int y) { return a[x] < a[y]; });
for (int i = 1; i <= n; ++i) id[rev[i]] = i;
for (int i = 1; i < n; ++i) sans += abs(a[i] - a[i + 1]);
for (int i = 2; i <= n; ++i) chsmin(ans[1], calc(1, i));
for (int i = 1; i < n; ++i) chsmin(ans[n], calc(i, n));
for (int i = 2; i <= n; ++i) {
chsmin(ans[i],
min({calc(1, i), calc(i - 1, i), calc(i, i + 1), calc(i, n)}));
}
for (int i = 2; i < n; ++i) {
s[i] = abs(a[i + 1] - a[i]) + abs(a[i] - a[i - 1]);
bd[i] = minmax(id[i - 1], id[i + 1]);
}
solve1();
solve2();
solve3();
solve4();
solve5();
solve6();
solve7();
solve8();
solve9();
for (int i = 1; i <= n; ++i) cout << ans[i] + sans << "\n";
return 0;
}
分类:
数据结构
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 地球OL攻略 —— 某应届生求职总结
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 提示词工程——AI应用必不可少的技术
· .NET周刊【3月第1期 2025-03-02】