【金凌试题】数列 可持久

题目描述

　　一个简单的数列问题：给定一个长度为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;
}

转载于:https://www.cnblogs.com/Cedric341561/p/6811015.html