poj 3928树状数组

来源:互联网 发布:excel数据统计图表 编辑:程序博客网 时间:2024/05/17 05:51
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
using namespace std;
#define ll long long
int dis[20010];
int b[20010];
int n;
void add(int x,int a)
{


    while(x<=n)
    {
        dis[x]+=a;//printf("ok\n");
       x+=x&(-x);
    }
}
int getsum(int x)
{
    int sum=0;
    while(x>0)
    {
        sum+=dis[x];
         x-=x&(-x);
    }
    return sum;
}
struct xiazou
{
    int val,pos;
}c[20010];
bool cmp(xiazou a,xiazou b)
{
    return a.val<b.val;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
       scanf("%d",&n);
       memset(dis,0,sizeof(dis));
       memset(b,0,sizeof(b));
       for(int i=1;i<=n;i++)
       {
            scanf("%d",&c[i].val);
            c[i].pos=i;
       }
       sort(c+1,c+n+1,cmp);
       for(int i=1;i<=n;i++)
        b[c[i].pos]=i;
        /*for(int i=1;i<=n;i++)
            printf("%d ",b[i]);*/
        long long sum=0;
        add(b[1],1);
       for(int i=2;i<=n;i++)
       {
           int low=getsum(b[i]);


            sum+=low*(n-b[i]-i+1+low)+(b[i]-1-low)*(i-1-low);//printf("%d %d..\n",sum,low);
            add(b[i],1);
       }
       printf("%lld\n",sum);
    }
}
0 0