USACO 2017 January Contest Gold T1: Balanced Photo

2025-06-01 05:34:07

世界杯足球球队排名

题目大意 FJ正在安排他的N头奶牛站成一排来拍照。(1 2 using namespace std; 3 const int MAXN=2e5+10; 4 5 struct Node{ 6 int h,id; 7 }a[MAXN]; 8 inline bool cmp(Node x,Node...

题目大意

FJ正在安排他的N头奶牛站成一排来拍照。(1<=N<=100,000)序列中的第i头奶牛的高度是h[i],且序列中所有的奶牛的身高都不同。

就像他的所有牛的照片一样,FJ希望这张照片看上去尽可能好。他认为,如果L[i]和R[i]的数目相差2倍以上的话,第i头奶牛就是不平衡的。(L[i]和R[i]分别代表第i头奶牛左右两边比她高的数量)。

如果L[i]和R[i]中较大者比较小者的数量严格多两倍的话,这头奶牛也是不平衡的。FJ不希望他有太多的奶牛不平衡。

请帮助FJ计算不平衡的奶牛数量。

题目分析

显然,一上来我们有一个对于每个数都扫一遍的 O(N2) 做法,考虑如何优化。

我们自然想到不能简单地按照原序列直接处理,所以先我们先将所有的数据从大到小去排列,其中第k大的数在原本的奶牛序列中的序号为i,

一个 f[i] 数组来表示现在第i大的数有无被进行以下的处理过,若已进行,则为1,否则为0。

然后我们按原本的数组顺序来枚举奶牛,处理完后,这头奶牛所对应的 f[i] 就置为1,这样就能够保证当前 f[i] 为1的奶牛在原序列中一定在 f[i] 为0的奶牛的左边,

所以当前奶牛 i 左边比他高的奶牛都已经处理过了,而数量就是 f[1]~f[i-1]中1的数量(比他大还比他前),然后可以通过减法得出 i 右边比他高的牛的数量。

那么这其实就是一个单点修改+区间查询,用树状数组来维护 f数组 即可。

1 #include

2 using namespace std;

3 const int MAXN=2e5+10;

4

5 struct Node{

6 int h,id;

7 }a[MAXN];

8 inline bool cmp(Node x,Node y){

9 return x.h>y.h;

10 }

11

12 int n,l,r,ans;

13 int b[MAXN],BIT[MAXN];

14 inline int lowbit(int x){

15 return x&(-x);

16 }

17 inline void Update(int x,int v){

18 for(int i=x;i<=n;i+=lowbit(i))

19 BIT[i]+=v;

20 }

21 inline int Query(int x){

22 int res=0;

23 for(int i=x;i;i-=lowbit(i))

24 res+=BIT[i];

25 return res;

26 }

27 int main(){

28 scanf("%d",&n);

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

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

31 a[i].id=i;

32 }

33 sort(a+1,a+n+1,cmp);

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

35 b[a[i].id]=i;

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

37 l=Query(b[i]);

38 r=b[i]-l-1;

39 if(l*2

40 Update(b[i],1);

41 }

42 printf("%d\n",ans);

43 return 0;

44 }