树状数组upc1976

1976: problem F

Time Limit: 3 Sec   Memory Limit: 64 MB
Submit: 69   Solved: 10
[ Submit][ Status][ Web Board]

Description

在一个小镇上住着n位武林高手，他们互相之间经常PK，不过PK的时候必须要有一位裁判在场。这个小镇上的房子从1—n依次排列（一条直线上），两个人要单挑时必须到另一个人的家里，让这个人作为裁判，裁判的武功不能同时比两个人低也不能同时比两个人高。由于每个高手都很懒，所以他们走的路程不能比他俩之间的距离远（即只能在他们序号之间高手中选择裁判）。求小镇上最多能进行几场PK。

Input

先输入一个整数T(T<=100)，包含T组测试数据，每组数据先输入n(n<=100000)，然后后面是n个数，表示n个武林高手的功夫ai,(ai<=100000)。

Output

能够进行的最大场数，每组数据占一行。

Sample Input

231 2 352 2 2 2 2

Sample Output

110

思路：跟poj1990有些相似。
下面是代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int MAX=500010;
struct node
{
    long long x,value;
}a[MAX];
long long n;
long long num[MAX];
bool cmp(node a,node b)
{
    return a.value<b.value;
}
long long sum(long long x)
{
    long long sum=0;
    while(x>0)
    {
        sum+=num[x];
        x-=(x&(-x));
    }
    return sum;
}
void update(long long x,int cnt)
{
    while(x<=n)
    {
        num[x]+=cnt;
        x+=(x&(-x));
    }
}
int main()
{
    //freopen("test2.in","r",stdin);
    int t;
    cin>>t;
    while(t--)
    {
        cin>>n;
        for(long long i=1;i<=n;i++)
        {
            scanf("%lld",&a[i].value);
            a[i].x=i;
        }
        memset(num,0,sizeof(num));
        stable_sort(a+1,a+n+1,cmp);
        long long ans=0;
        for(long long  i=1;i<=n;i++)
        {
            long long x=sum(a[i].x);
            ans+=x*((n-a[i].x)-(i-x-1))+(a[i].x-x-1)*(i-x-1);
            update(a[i].x,1);
        }
        cout<<ans<<endl;
    }
    return 0;
}