POJ 3928

来源：互联网发布：开源直播app源码编辑：程序博客网时间：2024/05/28 23:12

树状数组的应用。

把每个人当做裁判进行遍历，计算住在这个人之前的人中有多少个是比他弱的，记为pre_less[i],多少个人比他强的记为pre_more[i];同样的，住在他后面的人中，比他弱的记为suf_less[i],比他强的记为suf_more[i]

那么，很显然的对于每个裁判judger[i]，可以进行pre_less[i]*suf_more[i]+pre_more[i]+suf_less[i]场比赛，累加起来即得答案。

#include<cstdio>#define MAX_N 20000+5int n,a[MAX_N],c[100000+5],MAX_ai;int pre_more[MAX_N],pre_less[MAX_N],suf_more[MAX_N],suf_less[MAX_N];long long ans;int lowbit(int x){return x&(-x);}void add(int i,int val){    while(i<=MAX_ai)    {        c[i]+=val;        i+=lowbit(i);    }}int sum(int i){    int s=0;    while(i>0)    {        s+=c[i];        i-=lowbit(i);    }    return s;}int main(){int t;scanf("%d",&t);while(t--){scanf("%d",&n);MAX_ai=0;for(int i=1;i<=n;i++){scanf("%d",&a[i]);if(MAX_ai<a[i]) MAX_ai=a[i];}for(int i=1;i<=n;i++) pre_more[i]=0,pre_less[i]=0,suf_more[i]=0,suf_less[i]=0;for(int i=1;i<=MAX_ai;i++) c[i]=0;for(int i=1;i<=n;i++){add(a[i],1);pre_less[i]=sum(a[i]-1);pre_more[i]=i-1-pre_less[i];}for(int i=1;i<=MAX_ai;i++) c[i]=0;for(int i=n;i>=1;i--){add(a[i],1);suf_more[i]=sum(MAX_ai)-sum(a[i]);suf_less[i]=n-i-suf_more[i];}ans=0;for(int i=1;i<=n;i++){ans+=pre_less[i]*suf_more[i]+pre_more[i]*suf_less[i];}printf("%I64d\n",ans);}}

当然，从代码中不难看出，有两个数组是不必要开的，pre_more[]和suf_less[]是根据pre_less[]和suf_more[]算出来的，那么我们完全可以在最后运算时再计算……

#include<cstdio>#include<cstring>#define MAX_N 20000+5int n,a[MAX_N],c[100000+5],MAX_ai;int pre_less[MAX_N],suf_more[MAX_N];int lowbit(int x){return x&(-x);}void add(int i,int val){    while(i<=MAX_ai)    {        c[i]+=val;        i+=lowbit(i);    }}int sum(int i){    int s=0;    while(i>0)    {        s+=c[i];        i-=lowbit(i);    }    return s;}int main(){int t;scanf("%d",&t);while(t--){scanf("%d",&n);MAX_ai=0;for(int i=1;i<=n;i++){scanf("%d",&a[i]);if(MAX_ai<a[i]) MAX_ai=a[i];}for(int i=1;i<=n;i++) pre_less[i]=0,suf_more[i]=0;for(int i=1;i<=MAX_ai;i++) c[i]=0;for(int i=1;i<=n;i++){add(a[i],1);pre_less[i]=sum(a[i]-1);}for(int i=1;i<=MAX_ai;i++) c[i]=0;for(int i=n;i>=1;i--){add(a[i],1);suf_more[i]=sum(MAX_ai)-sum(a[i]);}long long ans=0;for(int i=2;i<=n-1;i++){ans+=pre_less[i]*suf_more[i]+(i-1-pre_less[i])*(n-i-suf_more[i]);}printf("%I64d\n",ans);}}

不难看到，程序在各方面都得到了小幅优化

0 0