您的位置:首页 > 其它

cogs 数列

2014-03-17 23:32 169 查看
树状数组入门级习题 。。。。。

题目链接http://218.28.19.228/cogs/problem/problem.php?pid=859

/*
一眼就看出来了就是逆序对的变种。在逆序对中,
树状数组c[a[i]]保存的是在a[i]之前的不大于a[i]的数的个数,
注意这里包括a[i本身,但这题呢不能等于,而且不包括本身。。
有个好的解决方法,就是先把每个数+1,在进行修改操作时对加1后的数
操作,但在“求和”操作时,要对原来的数操作。
画几下,就发现这样做保证了计的数都是比a[i]小的!

*/

#include<stdio.h>
#include<string.h>

#define MAXN 50010

int n,c[MAXN],f[MAXN],a[MAXN],g[MAXN];

int lowbit(const int &x)
{
return x&-x;
}

int getsum(const int &x)
{
int cnt=0;
for (int i=x;i>0;i-=lowbit(i))
cnt+=c[i];
return cnt;
}

void modify(const int &x,const int &delta)
{
for (int i=x;i<33000;i+=lowbit(i))
c[i]+=delta;
}

int main()
{
int i,x;
double ans=0;
freopen("queueb.in","r",stdin);
freopen("queueb.out","w",stdout);
scanf("%d",&n);

for (i=1;i<=n;++i)
{
scanf("%d",&a[i]);
a[i]+=1;
}

for (i=1;i<=n;++i)
{
modify(a[i],1);
f[i]=getsum(a[i]-1); //注意是a[i]-1,不是a[i]
}

memset(c,0,sizeof(c));
for (i=n;i>0;--i)
{
modify(a[i],1);
g[i]=getsum(a[i]-1);//注意是a[i]-1,不是a[i]
}

for (i=2;i<n;++i)
ans+=f[i]*g[i];
printf("%.lf\n",ans);
//while(1);

return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: