bzoj1484: [HNOI2009]通往城堡之路

来源:互联网 发布:cp交易网络是什么意思 编辑:程序博客网 时间:2024/04/30 15:29

传送门
这题太迷的。。。
并不会证这个玩意为什么是对的。。。
固定第一个点的值,其他点的值取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 longusing 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);    }}
0 0