cogs 数列

来源：互联网发布：iphone6s不能更新软件编辑：程序博客网时间：2024/05/21 09:59

树状数组入门级习题。。。。。

题目链接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 50010int 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;}

0 0