UVA 1428 - Ping pong 二叉索引树标准用法

分析:

  感觉白书写的不太好理解,说下自己稍微改动后的分析。

  想象一条数轴,数轴上的点表示能力值。

  用数组c[i]表示在二叉索引树中以a[i]结尾的“水平长条”。

  所以每输入一个能力值都要更新与之相关的c[i]的值。

  第i个人当裁判时,

  计算到目前为止(即a[1]到a[i-1])数轴上a[i]左边的标记的个数,即前缀和,存入front[i],则a[i]右边就有i-1-front[i]个。

  重新逆序计算到目前为止(即a[n]到a[i+1])数轴上a[i]左边的标记的个数,即前缀和,存入back[i],则a[i]右边就有n-i-back[i]。

  说的很乱。。意会就好了。。这是我的代码:

 

#include <stdio.h>

#include <string.h>



int t, n, a[20010], c[100010], front[20010], back[20010];



int lowbit(int x)

{

    return x & -x;

}



int sum(int x)

{

    int ans = 0;

    while(x > 0)

    {

        ans += c[x];

        x -= lowbit(x);

    }

    return ans;

}



void add(int x, int d)

{

    while(x <= 100000)

    {

        c[x] += d;

        x += lowbit(x);

    }

}



int main()

{

    scanf("%d", &t);

    while(t--)

    {

        scanf("%d", &n);

        memset(c, 0, sizeof(c));

        for(int i = 1; i <= n; i++)

        {

            scanf("%d", &a[i]);

            add(a[i], 1);

            front[i] = sum(a[i]-1);

        }

        memset(c, 0, sizeof(c));

        for(int i = n; i >= 1; i--)

        {

            add(a[i], 1);

            back[i] = sum(a[i]-1);

        }

        long long ans = 0;

        for(int i = 2; i < n; i++)

            ans += 1LL * front[i]*(n-i-back[i]) + (i-front[i]-1)*back[i];

        printf("%lld\n", ans);

    }

    return 0;

}

 

  

你可能感兴趣的:(ping)