【金凌试题】数列 可持久

题目描述

　　一个简单的数列问题：给定一个长度为n的数列，求这样的三个元素ai, aj, ak的个数，满足ai < aj > ak，且i < j < k。

题目大意

求区间有多少数比给定的vl小。

数据范围

(n <= 50000)

样例输入

5
1 2 3 4 1

样例输出

6

解题思路

应该是线段树（树状数组）的模板题。结果不知道为什么用了可持久权值线段树。不管了233。

代码

#include <bits/stdc++.h>using namespace std;inline int Getint(){int x=0,f=1;char ch=getchar();while('0'>ch||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while('0'<=ch&&ch<='9'){x=x*10+ch-'0';ch=getchar();}return x*f;}int vl[50005],rt[50005],cnt=0;struct node{    int L,r,son[2],Sum,mid;}Tree[1000005];void Build(int &v,int L,int r){    Tree[v=++cnt]=(node){L,r,0,0,0,(L+r)/2};    if(L==r)return;    Build(Tree[v].son[0],L,(L+r)/2);    Build(Tree[v].son[1],(L+r)/2+1,r);}void Insert(int &New,int Old,int val){    if(!Old)return;    Tree[New=++cnt]=Tree[Old];    Tree[New].Sum++;    if(val<=Tree[New].mid)Insert(Tree[New].son[0],Tree[Old].son[0],val);    else Insert(Tree[New].son[1],Tree[Old].son[1],val);}long long Ask(int New,int Old,int val){    if(!Old||val<=Tree[New].L)return 0;    if(Tree[New].r<val)return Tree[New].Sum-Tree[Old].Sum;    return Ask(Tree[New].son[0],Tree[Old].son[0],val)+Ask(Tree[New].son[1],Tree[Old].son[1],val);}long long Ans=0;int main(){    int n=Getint();    Build(rt[0],0,50001);    for(int i=1;i<=n;i++)vl[i]=Getint();    for(int i=1;i<=n;i++)Insert(rt[i],rt[i-1],vl[i]);    for(int i=2;i<n;i++)Ans+=Ask(rt[i-1],rt[0],vl[i])*Ask(rt[n],rt[i],vl[i]);    cout<<Ans<<"\n";    return 0;}