Ping pong UVA - 1428 树状数组

这个题目，就是对于一个树状数组的考察，更新，求和，先说一下题目的意思，题目的要求就是需要统一，满足要求的情况下，有多少中可能，我们需要做的就是枚举裁判，然后对于一个裁判，a0 在a1到ai-1比他小的有ci个，那就有i - ci个比他大的， 同样我们需要统计在ai+1 到an-1 有di个比ai小的，然后就会有 n - 1 - i - di 个比他大的，然后根据加法原理，得到的公式为(LL)(c[i]*(n - 1 - i - d[i])) + (LL)(d[i]*(i - c[i])); 然后O（n）的枚举就可以了。注意long
long

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <map>
#include <queue>
#include <math.h>
#include <stack>
#include <utility>
#include <string>
#include <sstream>
#include <cstdlib>
#include <set>
#define LL long long
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 100000 + 10;
int dir[4][2] = {{1,0},{0,1},{-1,0},{0,-1}};
int k[maxn];
int c[maxn];
int num[maxn];
int d[maxn];
int n;
int lowbit(int x)
{
return x & (-x);
}
void add(int x,int d)
{
while(x <= maxn)
{
k[x] += d;
x += lowbit(x);
}
}
int sum(int x)
{
int ans = 0;
while(x > 0)
{
ans += k[x];
x -= lowbit(x);
}
return ans;

}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
memset(num,0,sizeof(num));
memset(c,0,sizeof(c));
memset(d,0,sizeof(d));
memset(k,0,sizeof(k));
scanf("%d",&n);
for(int i = 0; i < n; i++)
{
scanf("%d",&num[i]);
}
for(int i = 0; i < n; i++)
{
add(num[i],1);
c[i] = sum(num[i] - 1);
}
memset(k,0,sizeof(k));
for(int i = n - 1; i >= 0; i--)
{
add(num[i],1);
d[i] = sum(num[i] - 1);
}
LL cnt = 0;
for(int i = 0; i < n; i++)
{
cnt += (LL)(c[i]*(n - 1 - i - d[i])) + (LL)(d[i]*(i - c[i]));
}
printf("%lld\n",cnt);
}
return 0;
}