分析:
感觉白书写的不太好理解,说下自己稍微改动后的分析。
想象一条数轴,数轴上的点表示能力值。
用数组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; }