UVA1428 - Ping pong(树状数组)

UVA1428 - Ping pong(树状数组)

题目链接

题目大意：有N个人，每个人都有一个技能值ai，现在要开展乒乓球比赛，要求要有两个选手和一个裁判，要求裁判需要在两名选手的中间而且技能值也是在两名选手的中间，问可以开展多少场比赛。

解题思路:对于第i个选手当裁判的话，设它前面位置的选手有ci个技能值比它低的，那么就有i - 1 - ci个比它高的，对于第i选手后面的位置，同样有di个技能值比它低的，那么就有N - i - di个比它高的。组合一下：ci∗(N
- i - di) + (i - 1 - ci)
∗
di.那么对于ci的值，根据i的位置，将Xi标为0或者1（在i位置前面就是1，后面就是0）。di类似求得。

代码：

#include <cstdio>
#include <cstring>

const int maxn = 1e5 + 5;
const int N = 2e4 + 5;

typedef long long ll;

int C[maxn];
int A[maxn];
ll c
, d
;

int lowbit(int x) { return x&-x; }

void Add (int x, int d) {

while (x < maxn) {

C[x] += d;
x += lowbit(x);
}
}

int Sum (int x) {

int ret = 0;
while (x > 0) {

ret += C[x];
x -= lowbit(x);
}
return ret;
}

void init () {

memset (C, 0, sizeof (C));
}

int main () {

int T;
int n;
int num
;
scanf ("%d", &T);
while (T--) {

scanf ("%d", &n);
init();
for (int i = 0; i < n; i++) {
scanf ("%d", &num[i]);
Add(num[i], 1);
c[i] = Sum (num[i]) - 1;
}

init();
for (int i = n - 1; i >= 0; i--) {

Add(num[i], 1);
d[i] = Sum (num[i]) - 1;
}

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