CodeForces - 978D·Almost Arithmetic Progression

题目

传送门

题目大意

给你一个长度为$n$的数列（$1\leq n\leq 100000$），对任意元素你可以最多进行一次操作：将其加$1$或减$1$，问将次数列变成等差数列的最小代价（当$n\leq 2$时，数列一定为等差数列）。

分析

比赛时脑袋抽，硬是想从数列相邻两数的差入手，结果凉了……
直接枚举前两个数的操作（共 9 9 <script type="math/tex" id="MathJax-Element-241">9</script>种），就知道了公差，然后就统计答案……

代码

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

#define MAXN 100000
#define INF 0x7fffffff
int N,Delta;
int B[MAXN+5],A[MAXN+5];

int Ans=INF;
void GetAns(int i,int st){
    memcpy(A,B,sizeof B);//由于后边有更改不方便还原，就直接拷贝一份
    while(i<=N){
        int tmp=Delta-(A[i-1]-A[i]);
        if(abs(tmp)>1)//只能更改A[i]，因为A[i-1]已经由A[i-2]固定了，所以tmp的范围只有[-1,1]
            return;
        A[i++]-=tmp;//自己举个例子就知道什么是减了
        st+=abs(tmp);//统计答案
    }
    Ans=min(Ans,st);
}

int main(){
    scanf("%d",&N);
    for(int i=1;i<=N;i++)
        scanf("%d",&B[i]);
    for(int i=-1;i<=1;i++)//第一个数的三种操作：+1、-1、不变
        for(int j=-1;j<=1;j++){//第二个数的三种操作
            B[1]+=i,B[2]+=j;
            Delta=B[1]-B[2];//公差
            GetAns(3,abs(i)+abs(j));//从第三个数开始
            B[1]-=i,B[2]-=j;
        }
    if(Ans==INF) puts("-1");
    else printf("%d",Ans);
}