atcoder C - ~

https://atcoder.jp/contests/abc406/tasks/abc406_c

题目简述：
给定一个序列p，让你求出p的所有子序列中波浪形序列的个数

波浪形序列的定义：1：长度>=4；2：仅存在一个波峰和波谷；3：序列的第一个数小于第二个数

#include<bits/stdc++.h>
using namespace std;
#define int long long
//typedef long long ll;
#define endl "\n"
//#define PII pair<int,int>
#define x first
#define y second
//priority_queue<int, vector<int>, greater<int>> pq;//小根堆
//{并查集
//int fa[N];
//int n;
//void init(){
//	for(int i=0;i<=n;i++)fa[i]=i;
//}初始化
//int get(int x){
//	return fa[x]=(fa[x]==x?x:get(fa[x]));
//}查找
//void merge(int a,int b){
//	fa[get(a)]=get(b);
//}合并
//}
const int N=5e5+10;
int a[N];
int n;
void solve(){
	cin >> n;
	for(int i=1;i<=n;i++)
	cin >>a[i];
	vector<pair<int,bool> >q;
	for(int i=1;i+2<=n;i++)
	{
		if(a[i+1]>a[i]&&a[i+1]>a[i+2])q.push_back(make_pair(i,true));
//波峰为true		
		if(a[i+1]<a[i]&&a[i+1]<a[i+2])q.push_back(make_pair(i,false));
//波谷为false		
	}
	int ans=0;
	for(int i=0;i+1<q.size();i++)
	if(q[i].y&&!q[i+1].y)
//相邻波峰波谷
	{
		int l=0,r=n-1;
		if(i>0)l=q[i-1].x;
		if(i+2<q.size())r=q[i+2].x;
		ans+=(q[i].x-l)*(r-q[i+1].x);
//计算所有可能位置
	}
	cout << ans << endl;
}	 
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int q=1;
//	cin >> q;
	while(q--)
	solve();
}