uvalive 4329 Ping pong (树状数组)

传送门：点击打开链接

题目大意：

给一个长度为N的数组，求有多少个三元组（i,j,k）满足（i<j<k && min(a[i],a[k]) < a[j] && max(a[i],a[k]) > a[j]）；

解题思路：

按值建数。那么可以求出 1~i-1之间有多少个数比a[i]小 (ans1[i]),那么就有i-1-ans1[i]个数比a[i]大。同理也可以求右边。

那么 以 i 为中间数。左小右大的个数就是ans1[i]*(n-i-ans2[i]),做大有小的个数就是ans2[i]*(i-1-ans1[i])。

#include <cstdio>
#include <cstring>
#define maxn 20010
#define maxm 100000
int C[maxm];
int lowbit(int x)
{
    return x&(-x);
}
int sum(int x)
{
    int res = 0;
    while(x > 0)
    {
        res += C[x];
        x -= lowbit(x);
    }
    return res;
}
void add(int x)
{
    while(x < maxm)
    {
        C[x]++;
        x += lowbit(x);
    }
}
int a[maxn];
int ans1[maxn],ans2[maxn];
int main()
{
    int T;
    scanf("%d",&T);
    for(int ks = 1;ks <= T;ks++)
    {
        int n;
        scanf("%d",&n);
        for(int i = 1;i <= n;i++)
            scanf("%d",&a[i]);
        memset(C,0,sizeof C);
        for(int i = 1;i <= n;i++)
        {
            add(a[i]);
            ans1[i] = sum(a[i]-1);
        }
        memset(C,0,sizeof C);
        for(int i = n;i >= 1;i--)
        {
            add(a[i]);
            ans2[i] = sum(a[i]-1);
        }

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