Bookface（中位数，保序回归）

#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;
}