问题 D: 数列-训练套题T10T3
时间限制: 4 Sec 内存限制: 128 MB
提交: 40 解决: 4
题目描述
数列(sequence.pas/c/cpp)
问题描述
一个简单的数列问题:给定一个长度为n的数列,求这样的三个元素ai, aj, ak的个数,满足ai < aj > ak,且i < j < k。
输入数据
第一行是一个整数n(n <= 50000)。
第二行n个整数ai(0 <= ai <= 32767)。
输出数据
一个数,满足ai < aj > ak (i < j < k)的个数。
样例输入
5
1 2 3 4 1
样例输出
6
经验总结
这一题,如果用动态规划(LIS)的方法求,会超时,这里使用的是树状数组,因为输入数据允许ai = 0 ,所以要注意0的处理,从左往右遍历求得序列中在元素左边比该元素小的元素个数,从右往左遍历求得序列中在元素右边比该元素小的元素个数,然后将两个数组相应元素相乘,再进行累加,最终得到的即为满足题目要求的序列的个数。
正确代码
#include<cstdio>
#include<cstring>
using namespace std;
#define lowbit(i) ((i)&(-i))
const int maxn=50010;
int c[maxn],sum1[maxn],sum2[maxn],a[maxn];
void update(int x,int v)
{
if(x==0)
c[0]+=v;
else
for(int i=x;i<maxn;i+=lowbit(i))
c[i]+=v;
}
int getsum(int x)
{
int sum=0;
if(x==-1)
return 0;
for(int i=x;i>0;i-=lowbit(i))
sum+=c[i];
return sum+c[0];
}
int main()
{
int n;
long long ans;
while(~scanf("%d",&n))
{
if(n==0)
break;
memset(c,0,sizeof(c));
for(int i=1;i<=n;++i)
{
scanf("%d",&a[i]);
update(a[i],1);
sum1[i]=getsum(a[i]-1);
}
memset(c,0,sizeof(c));
for(int i=n;i>=1;--i)
{
update(a[i],1);
sum2[i]=getsum(a[i]-1);
}
ans=0;
for(int i=1;i<=n;++i)
{
ans+=(sum1[i]*sum2[i]);
}
printf("%lld\n",ans);
}
}