BZOJ 3524: [Poi2014]Couriers 主席树题解

Time Limit: 20 Sec Memory Limit: 256 MB
Submit: 2314 Solved: 897

Description

给一个长度为n的序列a。1≤a[i]≤n。
m组询问，每次询问一个区间[l,r]，是否存在一个数在[l,r]中出现的次数大于(r-l+1)/2。如果存在，输出这个数，否则输出0。

Input

第一行两个数n，m。
第二行n个数，a[i]。
接下来m行，每行两个数l,r，表示询问[l,r]这个区间。

Output

m行，每行对应一个答案。

Sample Input

7 5

1 1 3 2 3 4 3

1 3

1 4

3 7

1 7

6 6

Sample Output

1

0

3

0

4

HINT

【数据范围】

n,m≤500000

2016.7.9重设空间，但未重测！

Source

By Dzy

---

题解：

查询区间是否有某数超过该区间数字个数的二分之一，这不是主席树的裸题嘛，让我来解释一波代码233

---

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<queue>
#include<set>
const int MAXN=10000010;
using namespace std;
int ls[MAXN],rs[MAXN],sum[MAXN],root[MAXN],n,m,xx,yy,sz;
//需要的数据结构是主席树,左子右子,该区间数的个数sum,以及树根的数组
void modify(int val,int x,int &y,int l,int r){
    y=++sz;sum[y]=sum[x]+1;//每次进来的时候,先给root[i]分配空间(也就是函数里的y)
    if(l==r) return;//如果分配到底层,不会再往下走了,所以return
    ls[y]=ls[x];rs[y]=rs[x];//先复制左右子树,然后再按情况更改左子树或者更新右子树
    int mid=(l+r)>>1;
    if(val<=mid) modify(val,ls[x],ls[y],l,mid);//看val是在左子树里还是右子树
    else         modify(val,rs[x],rs[y],mid+1,r);//右子树
}
int query(int from,int to){
    int l=1,r=n,mid,tmp=(to-from+1)>>1;//也就是我们要寻找的区间的二分之一大小
    int x=root[from-1],y=root[to];//x和y分别是from-1和to两个版本的主席树,由于前缀和的性质,我们即将可以进行加减来运算
    while(l!=r){//当l!=r时,一直二分处理
        if(sum[y]-sum[x]<=tmp)return 0;//如果该区间没有这么多数,那么直接让return 0;
        mid=(l+r)>>1;
        if(sum[ls[y]]-sum[ls[x]]>tmp){//看左区间有没有大于tmp个数,显然因为要求的是大于该区间的二分之一,所以不可能既左区间大于tmp,又右区间大于tmp
            r=mid;x=ls[x];y=ls[y];//往左寻找
        }else if(sum[rs[y]]-sum[rs[x]]>tmp){
            l=mid+1;x=rs[x];y=rs[y];//往右寻找
        }else return 0;
    }
    return l;//找到了就return l,否则在之前就return了0;
}

int main(){
    scanf("%d%d",&n,&m);
    for(register int i=1;i<=n;i++){
        scanf("%d",&xx);
        modify(xx,root[i-1],root[i],1,n);
    }
    for(register int i=1;i<=m;i++){
        scanf("%d%d",&xx,&yy);
        printf("%d\n",query(xx,yy));
    }
    return 0;
}