[HNOI2009]通往城堡之路

https://www.zybuluo.com/ysner/note/1251235

题面

给一个长度为(n)的序列({a})。现可改变第(2~n-1)个数的权值,使序列个数相差不超过(d)。最小化权值改变大小之和。

  • (nleq5000)

解析

留坑待填

这题很奇妙咦
无解显然是(abs(a[1]-a[n])>(n-1)*d)

注意到,最终序列(a')一定符合一个限制:(a'_i>=a[1]-(i-1)*d)
于是可以先把序列中所有值(除第一个)改为最小值,然后看怎么提高。

给一段区间同时升高(h)米。
为了方便,假设区间内(a_i)只有两种情况:(a_i+hleq b_i or a_igeq b_i)
( 其实这个情况很好保证,只要扫一遍取(h=min{b_i-a_i}))
如果第一种情况有(r)个,第二种情况有(s)个,则(ans-=(r-s)*h)
于是我们的目标就是使(r-s)(h)同号(h)可以为负)。

应用一下差分思想,发现可以用两个后缀区间表示一段区间。
于是就不用枚举右端点了。

然后可以发现,其实只要枚举左端点,在区间内取(h=min{b_i-a_i}),然后给所有区间同时升高(h),以此类推,总能使答案更优。
并且(b[n])只能升到(a[n]),不能变得更大。
则答案最优时,(a[n]=b[n])
当然也要注意修改的合法性,如(a_{l-1}+dgeq a_l)

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#define re register
#define il inline
#define ll long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define fp(i,a,b) for(re int i=a;i<=b;i++)
#define fq(i,a,b) for(re int i=a;i>=b;i--)
using namespace std;
const int mod=1e9+7,N=5050;
ll n,d,a[N],b[N],ans;
il ll gi()
{
   re ll x=0,t=1;
   re char ch=getchar();
   while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
   if(ch=='-') t=-1,ch=getchar();
   while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
   return x*t;
}
int main()
{
  re int T=gi();
  while(T--)
    {
      n=gi();d=gi();ans=0;
      fp(i,1,n) b[i]=a[i]=gi();
      if(abs(a[1]-a[n])>(n-1)*d) {puts("impossible");continue;}
      fp(i,2,n) b[i]=b[i-1]-d;
      while(a[n]^b[n])
    {
          re ll s=0,mn=1e18,mx=-1e18,pos,add=1e18;
          fq(i,n,2)
        {
          if(b[i]<a[i]) ++s,mn=min(mn,a[i]-b[i]);
          else --s;
          if(s>mx&&b[i-1]+d!=b[i]) mx=s,pos=i,add=mn;
        }
      add=min(add,b[pos-1]+d-b[pos]);
      fp(i,pos,n) b[i]+=add;
    }
      fp(i,1,n) ans+=abs(a[i]-b[i]);
      printf("%lld
",ans);
    }
  return 0;
}
原文地址:https://www.cnblogs.com/yanshannan/p/9483975.html