COGS 859 数列

859. 数列

★☆   输入文件： queueb.in   输出文件： queueb.out    简单对比
时间限制：1 s   内存限制：128 MB

【题目描述】

一个简单的数列问题：

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

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

<j<k。< span="">

【输入格式】

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

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

【输出格式】

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

<j<k）的个数。< span="">

【输入样例】

5
1
2
3
4
1

【输出样例】

6

【数据规模】

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

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

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

乍一看 找比它小的数的个数 似乎是个区间的问题

然后就套线段树了

这里有两种处理方法

1.

以每个数的最大值32767(32768)建树

维护sumv 即出现的次数

需要建立两个线段树

枚举中间点j

在左面的树中寻找比j小的数的个数

在右面的树中进行同样操作

然后结果相乘 再累加起来

这种适合处理A比较小的数据

等下会看到一个无法这样做的题目

先贴代码吧

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int lim=37777;
int sumv[2][lim<<2];
int m,n,a,b,c;
int s[50001];         //注意s存的是每个数 所以要开到50000 而不能开37777
long long l,r,z;

void pushup(int i,int o){sumv[i][o]=sumv[i][o<<1]+sumv[i][o<<1|1];}

int query(int i,int o,int l,int r,int w)
{
    if(r<=w)return sumv[i][o];

    int m=(l+r)>>1;
    int ret=0;
    if(w>m)ret+=query(i,o<<1|1,m+1,r,w);
    ret+=query(i,o<<1,l,m,w);
    return ret;
}

void add(int i,int o,int l,int r,int w,int k)
{
    if(l==r)
    {
        sumv[i][o]+=k;
        return;
    }
    int m=(l+r)>>1;
    if(w<=m)add(i,o<<1,l,m,w,k);
    else add(i,o<<1|1,m+1,r,w,k);
    pushup(i,o);
}

int main()
{
    freopen("queueb.in","r",stdin);
    freopen("queueb.out","w",stdout);
    m=35555;
    scanf("%d",&n);
    
    for(a=1;a<=n;a++)
    {
        scanf("%d",&s[a]);s[a]++;
        if(a==1)add(0,1,1,m,s[a],1);
        if(a>=3)add(1,1,1,m,s[a],1);
    }
    for(a=2;a<n;a++)
    {
		if(s[a]-1)
		{
			l=query(0,1,1,m,s[a]-1);
        	r=query(1,1,1,m,s[a]-1);
		}
		else l=r=0;
        z+=l*r;
        add(0,1,1,m,s[a],1);
        add(1,1,1,m,s[a+1],-1);
    }
    cout<<z<<endl;
    return 0;
}

因为这个建树和维护比较奇葩 所以调了好长时间 (-.-)

2.

以个数N建树

那么就只要找有多少比自己小的在自己前面 有多少比自己小的在自己后面 然后相乘

如何既维护比自己小 又维护序号呢？

按大小排序 从小到大来加入 保证之前的比自己小 那么查询值就都是比自己小的数了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int lim=50011;
int sumv[lim<<2];
int m,n,a,b,c;
struct self{int x,y;}s[lim];
long long l,r,z;

void pushup(int o){sumv[o]=sumv[o<<1]+sumv[o<<1|1];}

int w;
int querymin(int o,int l,int r)
{
    if(r<=w)return sumv[o];
    int m=(l+r)>>1;
    int ret=0;
    if(w>m)ret+=querymin(o<<1|1,m+1,r);
    ret+=querymin(o<<1,l,m);
    return ret;
}

int querymax(int o,int l,int r)
{
    if(l>=w)return sumv[o];
    int m=(l+r)>>1;
    int ret=0;
    if(w<=m)ret+=querymax(o<<1,l,m);
    ret+=querymax(o<<1|1,m+1,r);
    return ret;
}

void add(int o,int l,int r)
{
    if(l==r)
    {
        sumv[o]++;
        return;
    }
    int m=(l+r)>>1;
    if(w<=m)add(o<<1,l,m);
    else add(o<<1|1,m+1,r);
    pushup(o);
}

int cmp(self a1,self a2){return a1.y<a2.y;}

int q[lim],top;

int main()
{
    freopen("queueb.in","r",stdin);
    freopen("queueb.out","w",stdout);
    scanf("%d",&m);
    
    for(a=1;a<=m;a++)
    {
        scanf("%d",&s[a].y);
        s[a].x=a;
    }
    
    sort(s+1,s+m+1,cmp);
    
    for(a=1;a<=m;)
    {
		top=0;
		for(b=a;s[b].y==s[a].y;b++)
		{
			top++;
			q[top]=s[b].x;
			w=s[b].x-1;
			if(w>=1)l=querymin(1,1,m);else l=0;
			w=s[b].x+1;
			if(w<=m)r=querymax(1,1,m);else r=0;
			z+=l*r;
		}
		a=b;
		for(b=1;b<=top;b++)
		{
			w=q[b];
			add(1,1,m);
		}
	}
    cout<<z<<endl;
    return 0;
}