Problem - D - Codeforces
题意:
思路:
所谓枚举,就是要利用隐含条件找出不明显的小数据,即状态数很少
这道题中,因为是等差数列,且每个数只能被+1或-1一次
确定了前两个数和公差之后,数列就被确定了
而公差只有3种情况,前两个数只有9种情况
那么只需要O(9N)枚举即可
Code:
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int mxn=2e5+10;
int N;
int a[mxn],b[mxn];
int calc(int d){
memcpy(b,a,sizeof(b));
int res=0;
for(int i=3;i<=N;i++){
if(b[i]-b[i-1]>d+1||b[i]-b[i-1]<d-1) return 1e9;
if(b[i]-b[i-1]==d+1){
res++;
b[i]--;
}else if(b[i]-b[i-1]==d) continue;
else{
res++;
b[i]++;
}
}
return res;
}
void solve(){
cin>>N;
for(int i=1;i<=N;i++) cin>>a[i];
int D=a[2]-a[1],d=D;
int ans=1e9;
for(int i=-1;i<=1;i++){
for(int j=-1;j<=1;j++){
a[1]+=i;
a[2]+=j;
ans=min(ans,calc(a[2]-a[1])+(i!=0)+(j!=0));
a[1]-=i;
a[2]-=j;
}
}
if(ans==1e9) cout<<-1<<'\n';
else cout<<ans<<'\n';
}
signed main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int __=1;//cin>>__;
while(__--)solve();return 0;
}