HDU 2492 Ping pong

题目链接:HDU 2492 Ping pong

树状数组。

把技能值看成树状数组中的单元,所以要注意add函数中while循环的条件,是小于M而不是N,M是技能值的范围,N是人数。

先按输入顺序(也就是距离从一头到另一头的顺序)从左到右计算左边大的个数,左边小的个数,再从右到左计算右边大和小的个数,然后循环求交叉相乘的和。

注意hdu提交最好是用__int64,我用long long总是WA。

#include <iostream>
#include <stdio.h>
#include <cstring>

using namespace std;

const int M = 100005 + 10;
const int N = 20000 + 10;
int T,n;
int arr[N],l_l[N],r_l[N],num[M];

int lowbit(int i)
{
    return i & (-i);
}
int add(int i,int val)
{
    while(i < M)
    {
        num[i] += val;
        i += lowbit(i);
    }
}
int sum(int i)
{
    int ans = 0;
    while(i > 0)
    {
        ans += num[i];
        i -= lowbit(i);
    }
    return ans;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        memset(num,0,sizeof(num));
        scanf("%d",&n);
        for(int i = 1;i <= n;i++)
        {
            scanf("%d",&arr[i]);
            add(arr[i],1);
            l_l[i] = sum(arr[i] - 1);
        }
        memset(num,0,sizeof(num));
        for(int i = n;i > 0;i--)
        {
            add(arr[i],1);
            r_l[i] = sum(arr[i] - 1);
        }
        //long long s = 0;
        __int64 s = 0;
        for(int i = 1;i <= n;i++)
            s += (l_l[i] * (n - i - r_l[i]) + r_l[i] * (i - l_l[i] - 1));
        //printf ("%lld\n",s);
        printf("%I64d\n",s);
    }
    return 0;
}


你可能感兴趣的:(HDU 2492 Ping pong)