codeforce 481 div3 D. Almost Arithmetic Progression

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/A7_RIPPER/article/details/81737755

题目大意：

给你一个数列，每一项只能执行+1，-1或者不变（不算在操作次数内），最少操作多少次能使它变成一个等差数列，如果不能成为等差数列就输出-1。

解题思路：

首先根据前两项确定出公差然后模拟后面的情况即可。复杂度O(9n)，前两项对应9种情况，对符合要求的数列更新最小操作次数。
代码如下：

#include<iostream>
#include<cstdio>
#include<fstream>
#include<set>
#include<cmath>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<iomanip>
#include<cstdlib>
#include<list>
#include<queue>
#include<stack>
#include<algorithm>
#define inf 0x3f3f3f3f
#define MOD 1000000007
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define meminf(a) memset(a,inf,sizeof(a))
#define HASHP 13331;
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int a[100010];
int main()
{
  std::ios::sync_with_stdio(false);
  cin.tie(0);
  //freopen("test.txt","r",stdin);
  //freopen("output.txt","w",stdout);
  int n;
  cin>>n;
  for(int i=0;i<n;i++)cin>>a[i];
  if(n==1||n==2)cout<<0<<endl;
  else
  {
    int result=inf;
    for(int i1=-1;i1<=1;i1++)
    {
      int change1=0;
       int l0=a[0]+i1;
      if(i1!=0&&l0>=0)change1++;
      if(l0<0) continue;
      for(int i2=-1;i2<=1;i2++)
      {
        int l1=a[1]+i2,change2=change1;
        if(i2!=0&&l1>=0)change2++;
        if(l1<0) continue;
        int b=l1-l0,num=0;//确定公差
        bool flag=true;
        for(int j=2;j<n;j++)
        {
          int l3=a[j];
          if(l3-l1==b){l1=l3;continue;}
          else if((l3+1)==(b+l1))
          {
            num++;
            l3++;
            l1=l3;
          }
          else if((l3-1)==(b+l1))
          {
            num++;
            l3--;
            l1=l3;
          }
          else {flag=false;break;}
        }
        if(flag)//等差数列
        {
          int count=change2+num;
          result=min(count,result);//更新最小结果
        }
      }
    }
    if(result!=inf)cout<<result<<endl;
    else cout<<-1<<endl;
  }
  return 0;
}