UVALA 4329 - Ping pong 树状数组+组合原理

点击打开题目链接 UVALA 4329


题意与分析:

代码:

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;

const int MAXN = 20000 + 10;
const int MAXM = 100000 +10;
int num[MAXN];
int c[MAXN], d[MAXN], x[MAXM];

//x 的二进制表达式中最右边的1所对应的值
int lobit(int x)
{
    return x & -x;
}
//前缀和
int sum(int k)
{
    int ret = 0;
    while(k > 0)
    {
        ret += x[k];
        k -= lobit(k);
    }
    return ret;
}
//更新 x 数组
void add(int k)
{
    while (k <= 100000)
    {
        x[k] += 1;
        k += lobit(k);
    }
}

int main()
{
    int t, n;
    scanf("%d", &t);
    while (t--)
    {
        scanf("%d", &n);
        for (int i = 1; i <= n; i++)
        {
            scanf("%d", &num[i]);
        }
        memset(x, 0, sizeof(x));
        for (int i = 1; i <= n; i++)    //正序求c[i]
        {
            add(num[i]);
            c[i] = sum(num[i] - 1);
        }

        memset(x, 0, sizeof(x));
        for (int i = n; i >= 1; i--)    //逆序求d[i]
        {
            add(num[i]);
            d[i] = sum(num[i] - 1);
        }

        long long ans = 0;
        for (int i = 1; i <= n; i++)
        {
            ans += (long long)c[i] * (n - i - d[i]) + d[i] * (i - 1 - c[i]);
        }
        printf("%lld\n", ans);
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值