传送门
这题太迷的。。。
并不会证这个玩意为什么是对的。。。
固定第一个点的值,其他点的值取a[i]-(i-1)*d(这个点能取的最小值)
然后每次找一个后缀,把这个后缀所有数加一个大于0的值,且保证这是最优的取法下加的尽量大的值。
由于每次把一个数加到最值所以复杂度是O(n2)的。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#define N 5005
#define ll long long
using namespace std;
ll a[N],b[N],cas,n,d,inf,ma,val,num,pos,tar,ans;
int main(){
scanf("%lld",&cas);
while (cas--){
scanf("%lld%lld",&n,&d);
for (ll i=1;i<=n;i++) scanf("%lld",&a[i]);
if (abs(a[n]-a[1])>(n-1)*d){
printf("impossible\n");
continue;
}
b[1]=a[1];
for (ll i=2;i<=n;i++) b[i]=b[i-1]-d;
inf=1e18;
while (a[n]!=b[n]){
ma=-inf;
val=inf;
num=0;
for (ll i=n;i>1;i--){
if (b[i]<a[i]){
num++;
val=min(val,a[i]-b[i]);
}
else num--;
if (num>ma&&b[i]!=b[i-1]+d){
pos=i;
tar=val;
ma=num;
}
}
tar=min(tar,b[pos-1]+d-b[pos]);
for (ll i=pos;i<=n;i++) b[i]+=tar;
}
ans=0;
for (int i=1;i<=n;i++) ans+=abs(a[i]-b[i]);
printf("%lld\n",ans);
}
}