hrbust 哈理工oj 2042 序列问题【线段树+离散化】

序列问题

Time Limit: 3000 MS Memory Limit: 32768 K Total Submit: 146(74 users) Total Accepted: 85(67 users) Rating:  Special Judge: No

Description

计算序列 a[] 中， 当 i < j < k, a[i] < a[j]&&a[j] > a[k], （a[i]和a[k]无需比较大小）这样的子序列个数。

Input

多组输入数据： 每组数据第一行一个数n：代表序列长度为n（0<n<50000）； 接下来第二行有n个数：代表序列元素的值a[i](0<=a[i]<50000,且任意2个元素的值不相同)。

Output

对于每组数据输出符合条件的子序列的个数，每个输出占一行。

Sample Input

5 1 3 5 2 4

Sample Output

5

Source

HCPC2014校赛训练赛 1

第一发忘记离散化实力WA了一发，然后写上离散化，AC。

思路+解题过程：

操作次数很大，三层for估计会超时，不过理工的后台数据都比较水，并且题还给了3000ms，会不会超时我还真不知道，但是毕竟搞ACM的，尽量优化代码。

题目要求找出三元序列满足条件：a[i] < a[j]&&a[j] > a[k], 

我们可以for循环遍历一遍数组，从【0，i）找比a[i]小的数据，在（i，n-1】找比a[i]大的数据，然后相乘，就能得出包含i的三元序列的个数，遍历一遍后，加和即可、

我们两次建树和查找，第一次按位入树，找【0，i）找比a[i]小的数据，记录在aa【】数组里边，然后第二次建树 ，倒序入树，然后找【0，i）找比a[i]小的数据，记录在bb【】数组里边、

首先我们离散化：

        for(int i=0;i<n;i++)
        {
            scanf("%d",&da[i].date);
            da[i].pos=i+1;
        }
        sort(da,da+n,cmp);//排序
        for(int i=0;i<n;i++)//离散化
        {
           // printf("%d %d\n",da[i].pos,da[i].date);
            a[da[i].pos-1]=i+1;
        }

然后两次建树：

        for(int i=0;i<n;i++)//第一次正序然后找（0，i）比a【i】小的数据
        {
            aa[i]=Query(0,a[i],1,n,1);//先查找，再入树，查找的内容是从数据0，到数据a【i】有多少个数据比a【i】小
            update(a[i],1,1,n,1);
        }
        build(1,n,1);
        for(int i=n-1;i>=0;i--)//第二次逆序入树。
        {
            bb[i]=Query(0,a[i],1,n,1);//同理
            update(a[i],1,1,n,1);
        }
        ll output=0;
        for(int i=0;i<n;i++)
        {
            output+=aa[i]*bb[i];
        }
        printf("%lld\n",output);

完整的AC代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define lson l,m,rt*2
#define rson m+1,r,rt*2+1
#define ll long long int
using namespace std;
struct date
{
    int date,pos;
}da[50005];
int tree[500000];
int pos[500000];
int a[50005];
int aa[50005];
int bb[50005];
int cmp(date a,date b)
{
    if(a.date==b.date)
    return a.pos<b.pos;
    else
    return a.date<b.date;
}
void pushup(int rt)
{
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
void build( int l ,int r , int rt )
{
	if( l == r )
	{
		tree[rt]=0;
		return ;
	}
	else
	{
		int m = (l+r)>>1 ;
		build(lson) ;
		build(rson) ;
		pushup(rt) ;
	}
}
void update(int p,int c,int l,int r,int rt)//p阵营c数据.
{
    //printf("%d %d %d %d %d\n",p,c,l,r,rt);
    if(l==r)
    {
        tree[rt]+=c;
    }
    else
    {
        int m=(l+r)>>1;
        if(p<=m) update(p,c,lson);
        else  update(p,c,rson);
        pushup(rt);
		//printf("sum[%d]:%d\n",rt,tree[rt]);
    }
}
int Query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        return tree[rt];
    }
    else
    {
        int m=(l+r)>>1;
        int ans=0;
        if(L<=m)
        {
            ans+=Query(L,R,lson);
        }
        if(m<R)
        {
            ans+=Query(L,R,rson);
        }
        return ans;
    }
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        build(1,n,1);
        for(int i=0;i<n;i++)
        {
            scanf("%d",&da[i].date);
            da[i].pos=i+1;
        }
        sort(da,da+n,cmp);
        for(int i=0;i<n;i++)
        {
           // printf("%d %d\n",da[i].pos,da[i].date);
            a[da[i].pos-1]=i+1;
        }
        for(int i=0;i<n;i++)
        {
            aa[i]=Query(0,a[i],1,n,1);
            update(a[i],1,1,n,1);
        }
        build(1,n,1);
        for(int i=n-1;i>=0;i--)
        {
            bb[i]=Query(0,a[i],1,n,1);
            update(a[i],1,1,n,1);
        }
        ll output=0;
        for(int i=0;i<n;i++)
        {
            output+=aa[i]*bb[i];
        }
        printf("%lld\n",output);
    }
}