Poj 3828 Ping pong

最新推荐文章于 2026-06-02 09:20:07 发布

原创最新推荐文章于 2026-06-02 09:20:07 发布 · 857 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

Algorithm_树状数组

本文介绍了解决POJ 3928问题的方法，通过计算每个人作为裁判时的比赛数量来得出总的比赛数。利用树状数组实时求前缀和与后缀和，实现高效计算。

题目链接：http://poj.org/problem?id=3928

考虑第i个人当裁判的情形，假设a1到ai-1有ci个比ai小，那么就有(i-1)-ci个比ai大的；同理，假设ai+1 到an中有di个比ai小，那么就有n-i-di个比ai大。

这样最终i当裁判时有ci*(n-i-di) + (i-ci-1)*di中比赛。最后累加即可。

求ci的时候可以对计数下标进行树状数组实时求前缀和。关于后缀和di,可以逆序求解。

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <time.h>
#include <vector>
#include <set>
#include <map>
#include <algorithm>

using namespace std;
#define Maxn 100005
int x[Maxn];//
int a[20005];//原始数组
int c[20005];//前缀和
int d[20005];//后缀和
int lowbit(int y)
{
    return y&(-y);
}
int sum(int y)
{
    int result = 0;
    while(y>0)
    {
        result += x[y];
        y -= lowbit(y);
    }
    return result;
}
int add(int y,int s)
{
    while(y<=Maxn-1)
    {
        x[y] += s;
        y += lowbit(y);
    }
}
int main()
{

#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
#endif
    int t;
    int n;
    long long ans = 0;
    scanf(" %d",&t);
    while(t--)
    {
        ans = 0;
        scanf(" %d",&n);
        memset(x,0,sizeof(x));
        for(int i=1;i<=n;i++)
        {
            scanf(" %d",&a[i]);
            c[i] = sum(a[i] - 1);
            add(a[i],1);
        }
        memset(x,0,sizeof(x));
        for(int i=n;i>=1;i--)
        {
            d[i] = sum(a[i] - 1);
            add(a[i],1);
        }
        for(int i=1;i<=n;i++)
        {
            ans += (long long)c[i] * (n-i-d[i]) + (long long)(i-1-c[i])*d[i];
        }
        printf("%lld\n",ans);
    }

}