【枚举】CF978 D

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;
}