#include<bits/stdc++.h>
#define ll long long
using namespace std;
/*
链接 :https://codeforces.com/gym/102576
题意 :给定 a 数组, 可以花费一个代价对 a[i] 加减一,求 abs(a[i] - a[j]) >= d 的最小代价
分析 :先对 a 数组排序,将问题转化为 a[i + 1] - a[i] >= d 的最小代价
对 a[i] -= i * d,将问题转化为 a[i + 1] - a[i] >= 0
也就是说要我构造一个不递减的数组 b 使得 sum(abs(a[i] - b[i])) 最小
这是经典问题
若 a[1] <= a[2] <= ... <= a[n],则让 b[i] = a[i] 即可
若 a[1] > a[2] > ... > a[n] ,则让 b[i] = a 的中位数即可
考虑将 a 分成多个区间,每一个都是降序的区间
如果第 i 个区间的中位数小于等于第 i + 1 个位置的中位数则两个区间没有影响
反之,则需要合并两个区间并重新找中位数
区间中位数使用主席树维护
*/
const int N = 2e5 + 10;
int n, m, d;
ll a[N], b[N];
struct node{
int ls, rs, sum;
}tree[N<<5];
int rt[N], sz;
void update(int l, int r, int &x, int y, int p)
{
x = ++sz;
tree[x] = tree[y];
tree[x].sum++;
if(l == r) return;
int mid = (l + r) / 2;
if(p <= mid) update(l, mid, tree[x].ls, tree[y].ls, p);
else update(mid + 1, r, tree[x].rs, tree[y].rs, p);
}
int query(int l, int r, int x, int y, int k)
{
int temp = tree[tree[y].ls].sum - tree[tree[x].ls].sum;
if(l == r) return l;
int mid = (l + r) / 2;
if(temp >= k) return query(l, mid, tree[x].ls, tree[y].ls, k);
else return query(mid+1, r, tree[x].rs, tree[y].rs, k - temp);
}
void solve()
{
cin >> n >> d;
for(int i = 1; i <= n; i++) cin >> a[i];
sort(a + 1, a + 1 + n);
for(int i = 1; i <= n; i++) a[i] -= 1ll * i * d, b[i] = a[i];
sort(b + 1, b + 1 + n);
int len = unique(b + 1, b + 1 + n) - b - 1;
for(int i = 1; i <= n; i++)
{
int p = lower_bound(b + 1, b + 1 + len, a[i]) - b;
a[i] = p;
update(1, len, rt[i], rt[i-1], p);
}
stack<array<ll, 2>> stk;
for(int i = 1; i <= n; i++)
{
if(stk.empty() || a[i] >= stk.top()[0]) stk.push({a[i], i});
else
{
int now = a[i], l = i;
while(stk.size() && now < stk.top()[0])
{
l = stk.top()[1];
now = query(1, len, rt[l - 1], rt[i], (i - l + 2) / 2);
stk.pop();
}
stk.push({now, l});
}
}
vector<ll> ans;
int r = n;
while(stk.size())
{
auto [x, l] = stk.top();
stk.pop();
while(r >= l)
{
ans.push_back(b[x]);
r--;
}
}
reverse(ans.begin(), ans.end());
for(int i = 0; i < n; i++)
{
if(ans[i] + 1ll * (i + 1) * d < 0) ans[i] = 1ll * -(i + 1) * d;
if(i && ans[i] < ans[i - 1]) ans[i] = ans[i - 1];
}
ll res = 0;
for(int i = 1; i <= n; i++) res += abs(b[a[i]] - ans[i - 1]);
cout << res << '\n';
for(int i = 0; i <= sz; i++) tree[i] = {0, 0, 0};
sz = 0;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
int t = 1;
cin >> t;
while(t--) solve();
return 0;
}
Bookface(中位数,保序回归)
最新推荐文章于 2024-09-25 13:54:13 发布