[P3453] [POI2007] DRZ - Trees

毫无技术含量的题目。

首先,特判 1n 的答案,还有相邻的贡献。

然后记 li=min(ai1,ai+1),记 ri=max(ai1,ai+1),记 si=|aiai1|+|aiai+1|,考虑 ij 所作贡献(i,j[2,n],|ij|>1):

(1)ai<ljaj<li

δ=li+ri2aisi+lj+rj2ajsj

(2)ljai<rjaj<li

δ=li+risi+rjlj2ajsj

(3)airjaj<li

δ=li+ri+2aisiljrj2ajsj

(4)ai<ljliaj<ri

δ=rili2aisi+lj+rjsj

(5)ljai<rjliaj<ri

δ=rilisi+rjljsj

(6)airjliaj<ri

δ=rili+2aisiljrjsj

(7)ai<ljajri

δ=liri2aisi+lj+rj+2ajsj

(8)ljai<rjajri

δ=lirisi+rjlj+2ajsj

(9)airjajri

δ=liri+2aisiljrj+2ajsj

j 看作修改,把 i 看作询问。

j 对应的修改看成 (l,aj)(r,aj) 的线段(限制为 lair)。

i 对应的询问看成 (ai,l)(ai,r) 的线段(限制为 lajr)。

两个线段有交即产生贡献,直接对 x 轴扫描线即可。

为了满足 |ij|>1,所以首先离散化的时候不应该去重,然后询问的时候特判 ai1ai+1 两个纵坐标不计算贡献。ai 计算贡献无所谓,因为其贡献为 0

#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;
}
posted @   JCY_std  阅读(57)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 地球OL攻略 —— 某应届生求职总结
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 提示词工程——AI应用必不可少的技术
· .NET周刊【3月第1期 2025-03-02】
点击右上角即可分享
微信分享提示