首页 > 其他 > 详细

cf 61E. Enemy is weak 树状数组求逆序数

时间:2014-01-21 15:32:46      阅读:358      评论:0      收藏:0      [点我收藏+]

求出每个位置左边有几个比它大,右边有几个比它小,然后乘法原理加起来就够了。

大于小于什么的用树状数组YY一下就出来了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<map>
#include<iostream>
using namespace std;
int n;
int c[2000000];
struct node
{
    int x;
    int id;
}a[2000000];
bool cmp(node x,node y)
{
    return x.x<y.x;
}
int mp[2000000];
int ls[2000000];
int lb[2000000];
long long rs[2000000];
int lowbit(int x)
{
     return x&(-x);
}
int Sum(int n)
{
    int sum=0;
    while(n>0)
    {
         sum+=c[n];
         n=n-lowbit(n);
    }
    return sum;
}
void change(int i,int x)
{
     while(i<=n)
     {
          c[i]=c[i]+x;
          i=i+lowbit(i);
     }
}
int main()
{
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i].x);
            a[i].id=i;
        }
        sort(a+1,a+1+n,cmp);
        for(int i=1;i<=n;i++) mp[a[i].id]=i;

        for(int i=1;i<=n;i++)
        {
            change(mp[i],1);
            ls[mp[i]]=Sum(mp[i])-1;
            lb[mp[i]]=i-ls[mp[i]]-1;
        }
        long long ans=0;
//        for(int i=1;i<=n;i++)
//        {
//            printf("%d ",lb[mp[i]]);
//        }
//        puts("");
//        for(int i=1;i<=n;i++)
//        {
//            printf("%d ",ls[mp[i]]);
//        }
        for(int i=1;i<=n;i++)
        {
            rs[mp[i]]=(mp[i]-1-ls[mp[i]]);
            ans+=rs[mp[i]]*lb[mp[i]];
        }
        cout<<ans<<endl;
    return 0;
}


cf 61E. Enemy is weak 树状数组求逆序数

原文:http://blog.csdn.net/t1019256391/article/details/18559019

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!