题目大意
有一个台灯,等级为1~m的一个整数。
有一个喜爱等级x,你每次操作可以让台灯等级+1(m+1会变成1)或变成x。
现在你要完成台灯等级从ai变成ai+1这n-1个操作。
如何选择x能使操作次数最小。
做法
每次可以讨论出某些区间需要变成x然后到达ai+1比较优还是直接加到ai+1比较优。
然后打上对应前缀tag,最后顺序扫一遍找一个最小的。
#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=100000+10;
ll sum[maxn],ans,wdc;
int num[maxn],a[maxn];
int i,j,k,l,t,n,m,x,y;
void change(int l,int r,int v){
sum[l]+=(ll)v;
sum[r+1]-=(ll)v;
}
void modify(int l,int r,int v){
num[l]+=v;
num[r+1]-=v;
}
int main(){
scanf("%d%d",&n,&m);
fo(i,1,n) scanf("%d",&a[i]);
fo(i,1,n-1){
x=a[i];y=a[i+1];
if (x==y) continue;
if (x<y){
change(1,x,y-x);
change(y+1,m,y-x);
change(x+1,y,y+1);
modify(x+1,y,-1);
}
else{
change(y+1,x,y+m-x);
change(x+1,m,y+m+1);
modify(x+1,m,-1);
change(1,y,y+1);
modify(1,y,-1);
}
}
ans=100000000000000;
fo(i,1,m){
sum[i]+=sum[i-1];
num[i]+=num[i-1];
wdc=sum[i]+(ll)num[i]*i;
ans=min(ans,wdc);
}
printf("%lld\n",ans);
}