poj 3928 Ping pong（离散化+树状数组）

 

题目大意：N个乒乓球队员住在从西向东的街道旁，每个队员都有一个唯一的排名，为了提高他们的排名，队员之间要进行比赛，比赛时需要一个裁判，对裁判的选择有两个要求。

1.       裁判的排名不能高于两个比赛队员的排名或者低于两个比赛队员的排名，换句话说，就是裁判的排名要在两个比赛队员排名之间。

2.       由于队员们都比较lazy，而且比赛必须去裁判的家中，所以对于裁判的地理位置有个要求，两个比赛队员所走的距离之和不能大于两个队员的相隔距离。及裁判必须住在这两个队员之间。

问题：满足上述两个条件的比赛，能够举行多少场。

方法：

从裁判入手，就变成了从裁判左边找一个队员，右边找一个队员，有两种情况。

1.       左边队员的rank>裁判的rank，右边的队员rank<裁判的rank。

2.       左边队员的rank<裁判的rank，右边的队员rank>裁判的rank。

第i个队员作为裁判能够举行多少场呢？

从i前的i-1个队员中选择rank比他小的队员，从i后的N-i个队员中选择rank比他大的队员。

 

从i前的i-1个队员中选择rank比他大的队员，从i后的N-i个队员中选择rank比他小的队员。

使用两次树状数组进行统计。

1.       for(i=1--àN)

对于i前i-1个队员，统计rank比i小的队员个数和rank比i大的队员个数。使用Left数组记录。

 

2.       for(N-à1)

对于i后面N-i个队员中，统计rank比i小的队员个数和rank比i大的队员个数。使用Right数组记录。

队员i作为裁判能够举行的比赛场数为left[i].low*right[i].high+left[i].high*right[i].low

 

另外：使用离散化来优化队员的排名，结果必须使用long long 保存，但是我不知道为什么结果为什么必须使用long long,如果有明白的朋友，可以给我留言交流。

#include <iostream> using namespace std; #define MAX 20002 typedef struct RESULT { int high; int low; }; RESULT Left[MAX],Right[MAX]; int tree[MAX];//树状数组统计使用的数组 int rank[MAX]; int tag[MAX*5]; int N,Max=0; int lowbit(int x) { return x&(-x); } void add(int x,int val) { while(x<=N) { tree[x]+=val; x+=lowbit(x); } return; } int Sum(int x) { int s=0; while(x>0) { s+=tree[x]; x-=lowbit(x); } return s; } void discretization() { int cn=0; for(int i=1;i<=Max;i++) if(tag[i]!=0) { rank[tag[i]]=++cn; } } int main() { int T; scanf("%d",&T); while(T--) { int i; long long ans=0;//必须使用long long Max=0; memset(tree,0,sizeof(tree)); memset(tag,0,sizeof(tag)); scanf("%d",&N); for(i=1;i<=N;i++) { scanf("%d",&rank[i]); tag[rank[i]]=i; if(Max<rank[i]) Max=rank[i]; } discretization();//离散化排名 for(i=1;i<=N;i++) { Left[i].high=0; Left[i].low=0; Right[i].high=0; Right[i].low=0; } //统计i左边的情况 for(i=1;i<=N;i++) { int ans2=Sum(rank[i]); add(rank[i],1); Left[i].high=i-1-ans2; Left[i].low=ans2; } memset(tree,0,sizeof(tree)); //统计i右边的情况 for(i=N;i>0;i--) { int ans2=Sum(rank[i]); add(rank[i],1); Right[i].high=N-i-ans2; Right[i].low=ans2; } for(i=1;i<=N;i++) { ans+=1LL*Left[i].high*Right[i].low; ans+=1LL*Left[i].low*Right[i].high; } printf("%lld/n",ans); } return 0; }