P1970 花匠

题目描述

花匠栋栋种了一排花，每株花都有自己的高度。花儿越长越大，也越来越挤。栋栋决定把这排中的一部分花移走，将剩下的留在原地，使得剩下的花能有空间长大，同时，栋栋希望剩下的花排列得比较别致。

具体而言，栋栋的花的高度可以看成一列整数h1​,h2​,...,hn​。设当一部分花被移走后，剩下的花的高度依次为g1​,g2​,...,gm​，则栋栋希望下面两个条件中至少有一个满足：

条件 AA：对于所有g2i​>g2i−1​,g2i​>g2i+1​

条件 BB：对于所有g2i​<g2i−1​,g2i​<g2i+1​

注意上面两个条件在m=1时同时满足，当m>1时最多有一个能满足。

请问，栋栋最多能将多少株花留在原地。

输入格式

第一行包含一个整数n，表示开始时花的株数。

第二行包含n个整数，依次为h1​,h2​,...,hn​,表示每株花的高度。

输出格式

一个整数m，表示最多能留在原地的花的株数。

输入输出样例

输入 #1

5
5 3 2 1 2

输出 #1

3

说明/提示

【输入输出样例说明】

有多种方法可以正好保留 3 株花，例如，留下第 1、4、5 株，高度分别为 5、1、2，满足条件 B。

【数据范围】

对于 20%的数据，n≤10；

对于 30%的数据，n≤25；

对于 70%的数据，n≤1000,0≤hi​≤1000；

对于 100%的数据，1≤n≤100,000,0≤hi​≤1,000,000，所有的hi​随机生成，所有随机数服从某区间内的均匀分布。

思路

一直下降突然上升ans++，一直上升突然下降ans++，ans初值赋为1。

代码：

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int N=100010;

int n,h[N];
int flag,ans;

int main() {
	scanf("%d",&n);
	for(int i=1; i<=n; i++)
		scanf("%d",&h[i]);
	if(h[2]>=h[1])
		flag=1;
	for(int i=1; i<=n; i++) {
		if(flag==0&&i==n) {
			ans++;
			break;
		}
		if(flag==1)
			if(h[i+1]<h[i]) {
				ans++;
				flag=0;
				continue;
			}
		if(flag==0)
			if(h[i+1]>h[i]) {
				ans++;
				flag=1;
				continue;
			}
	}
	ans++;
	printf("%d\n",ans);
	return 0;
}