题目传送门
题意: 给你一个数组,你可以对数组中每一个元素进行+1,-1一次,或者不操作,问你是否能使这个数组变成等差数列,如果能请打印最少操作次数,如果不能打印-1。
1≤n≤100000
思路: 等差数列就有一个公差,我们可以枚举前两个数,一共9种操作,前两个数确定之后,这个数列公差就确定了,然后我们从3开始遍历,看每个数是否能通过操作或者不操作达到合法,维护最小值即可。
#include<bits/stdc++.h>
using namespace std;
const int N=2e5+5;
const int inf=0x7fffffff;
const int mod=1e9+7;
const int eps=1e-6;
typedef long long ll;
typedef unsigned long long ull;
#define ls p<<1
#define rs p<<1|1
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define pii pair<int,int>
#define int long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define endl '\n'
#define null NULL
int a[N],b[N];
signed main()
{
IOS;
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
b[i]=a[i];
}
if(n<=2)
{
cout<<0<<endl;
return 0;
}
int res=inf;
for(int i=a[1]-1;i<=a[1]+1;i++)
{
for(int j=a[2]-1;j<=a[2]+1;j++)
{
int d=j-i;int cnt=0;
if(i!=a[1])
cnt++;
if(j!=a[2])
cnt++;
for(int k=3;k<=n;k++)
{
int p=a[k-1];
if(k==3)
p=j;
if(abs(p+d-a[k])<=1)
{
if(p+d-a[k]!=0)
cnt++;
a[k]=p+d;
}
else
{
cnt=inf;
break;
}
}
res=min(res,cnt);
for(int k=3;k<=n;k++)
{
a[k]=b[k];
}
}
}
if(res==inf)
{
cout<<-1<<endl;
}
else
{
cout<<res<<endl;
}
}