Codeforces Round #695 (Div. 2) B. Hills And Valleys

传送门

题目大意

如果a_i-1 > a_i && a_i < a_i+1那么就会形成一个山谷，如果a_i-1 < a_i && a_i > a_i+1那么就会形成一个山顶，我们可以随意的将其中一个数改成任意值，找出山谷和山顶总量和的最小值

题解

我们先用一个数组来记录哪个是山顶或山谷，然后一个一个试着改他们就好了，当我们改了i之后，i-1、i+1为中心的山顶或山谷都有可能改变，所以要重新算一下就好了

#include<cstdio>
#include<algorithm>
using namespace std;
const int MAX=3e5+5;
int a[MAX];
int is[MAX];
int n;
int isHill(int i){
	if(i<=0||i>=n-1){
		return 0;
	}
	return a[i-1]<a[i]&&a[i]>a[i+1];
}
int isValley(int i){
	if(i<=0||i>=n-1){
		return 0;
	}
	return a[i-1]>a[i]&&a[i]<a[i+1];
}
int main(){
	int t;
	scanf("%d",&t);
	while(t--){
		scanf("%d",&n);
		for(int i=0;i<n;i++){
			scanf("%d",&a[i]);
			is[i]=0;
		}
		int ans=0;
		for(int i=1;i<n-1;i++){
			if(isHill(i)){
				ans++;
				is[i]=1;
			}else if(isValley(i)){
				ans++;
				is[i]=1;
			}
		}
		int keep=ans;
		for(int i=1;i<n-1;i++){
			int temp=a[i];
			a[i]=a[i-1];
			ans=min(ans,keep-is[i]-is[i-1]-is[i+1]+isHill(i)+isHill(i-1)+isHill(i+1)+isValley(i)+isValley(i-1)+isValley(i+1));
			a[i]=a[i+1];
			ans=min(ans,keep-is[i]-is[i-1]-is[i+1]+isHill(i)+isHill(i-1)+isHill(i+1)+isValley(i)+isValley(i-1)+isValley(i+1));
			a[i]=temp;
		}
		printf("%d\n",ans);
	}
}