cogs 859 数列【线段树】

【题目描述】

一个简单的数列问题：

给定一个长度为n的数列，求这样的三个元素 ai,aj,ak 的个数，

满足 ai<aj>ak，且 i<j<k。

<j<k。< span="" style="font-family: serif; font-size: 16px; line-height: 20px;">

【输入格式】

第1行是一个整数n（1<=n<=50000）。

接下来n行，每行一个元素ai（0<=ai<=32767）。

【输出格式】

一个数，满足 ai<aj>ak (i<j<k) 的个数。

<j<k）的个数。< span="" style="font-family: serif; font-size: 16px; line-height: 20px;">

【输入样例】

512341

【输出样例】

6

【数据规模】

对于30%的输入数据有n<=200。

对于80%的输入数据有n<=10000。

对于100%的输入数据有n<=50000。

题解：

因为a只有32767，所以我们可以把a的值当做下标。等于把下标为a的点染上颜色。。。线段树中的值保存染上颜色的点的个数

这样我们查询区间和即可。。

#include<iostream>#include<cstdio>#include<cstring>using namespace std;long long tree[200000]={0},minl[50001]={0},minr[50001]={0},a[50001]={0};void insert(int i,int l,int r,int aa){    int mid;    if (l==r&&r==aa)    {        ++tree[i];        return;    }    mid=(l+r)/2;    if (aa<=mid) insert(i*2,l,mid,aa);    else insert(i*2+1,mid+1,r,aa);    tree[i]=tree[i*2]+tree[i*2+1];}long long work(int i,int l,int r,int aa,int b){    long long ans=0;    int mid;    if (aa<=l&&r<=b)        return tree[i];    mid=(l+r)/2;    if (aa<=mid) ans+=work(i*2,l,mid,aa,b);    if (b>mid) ans+=work(i*2+1,mid+1,r,aa,b);    return ans;}int main(){    freopen("queueb.in","r",stdin);    freopen("queueb.out","w",stdout);    int n,i,j;    long long ans=0;    scanf("%d",&n);    for (i=1;i<=n;++i)    {      scanf("%lld",&a[i]);      a[i]+=2;    }    for (i=1;i<=n;++i)    {        insert(1,1,32769,a[i]);        if (i>1)          minl[i]=work(1,1,32769,1,a[i]-1);    }    memset(tree,0,sizeof(tree));    for (i=n;i>=1;--i)    {        insert(1,1,32769,a[i]);        if (i<n)          minr[i]=work(1,1,32769,1,a[i]-1);        if (i>1)          ans+=minl[i]*minr[i];    }        printf("%lld\n",ans);        fclose(stdin);    fclose(stdout);}