序列问题 | ||||||
| ||||||
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);
}
}