HDU 2492 Ping pong

最新推荐文章于 2019-09-27 21:03:01 发布

原创最新推荐文章于 2019-09-27 21:03:01 发布 · 528 阅读

0 ·

CC 4.0 BY-SA版权

数据结构与算法专栏收录该内容

338 篇文章

订阅专栏

本文介绍了解决HDU2492 Pingpong问题的方法，利用树状数组来处理技能值的计算。文章详细展示了如何通过两次遍历计算左右两侧的大于或小于当前值的数量，并通过交叉相乘求和得到最终答案。

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