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