bzoj3524: [Poi2014]Couriers
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
解法一、
还记得区间第k小数怎么查询吗?http://www.cnblogs.com/TheRoadToTheGold/p/6259781.html
如果左子节点新增的数的个数<=k,递归左子节点,反之递归右子节点,k变为k-左子节点新增的数的个数
这里是查询区间出现次数>k的数
与区间第k小一样,都应用了数出现的次数
所以只需要在查询时,如果左子节点新增数的个数>k,递归左子节点;如果右子节点新增数的个数>k,递归右子节点;再否则,那就没有满足条件的数
答案就是递归到的叶子节点
#include<cstdio> #include<algorithm> #define N 500001 using namespace std; int n,m,tot,a[N],hash[N]; int root[N],sum[N*20],lc[N*20],rc[N*20],cnt; int ans; void discrete() { sort(a+1,a+n+1); tot=unique(a+1,a+n+1)-(a+1); for(int i=1;i<=n;i++) hash[i]=lower_bound(a+1,a+tot+1,hash[i])-a; } inline void insert(int pre,int & now,int l,int r,int k) { sum[now=++cnt]=sum[pre]+1; if(l==r) return; int mid=l+r>>1; if(k<=mid) { rc[now]=rc[pre]; insert(lc[pre],lc[now],l,mid,k); } else { lc[now]=lc[pre]; insert(rc[pre],rc[now],mid+1,r,k); } } inline void query(int x,int y,int l,int r,int k) { if(l==r) {ans=a[l];return;} int mid=l+r>>1; if(sum[lc[y]]-sum[lc[x]]>k) query(lc[x],lc[y],l,mid,k); else if(sum[rc[y]]-sum[rc[x]]>k) query(rc[x],rc[y],mid+1,r,k); else {ans=0;return ;} } int main() { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) scanf("%d",&a[i]),hash[i]=a[i]; discrete(); for(int i=1;i<=n;i++) insert(root[i-1],root[i],1,tot,hash[i]); int s,t,k; for(int i=1;i<=m;i++) { scanf("%d%d",&s,&t); k=t-s+1>>1; query(root[s-1],root[t],1,tot,k); printf("%d\n",ans); } }
解法二、
记得中庸之道那道题吗?http://www.cnblogs.com/TheRoadToTheGold/p/6291149.html
一个区间的中位数是第(r-l)/2+1小,所以如果查找是否有数出现了(r-l)/2+1次,如果存在这个数,那么排序后中位数的位置一定是这个数,所以直接查中位数的出现次数就好了
这里是查找是否有数出现了大于(r-l+1)/2次,即至少出现了(r-l+1)/2+1次
由于整除,所以
如果区间有奇数个点,(r-l+1)/2+1=(r-l)/2+1=区间中位数的位置
如果区间有偶数个点,(r-l+1)/2是把区间从中间劈开,左边一部分的最后一个位置,那么+1后就成为了右边一部分的第一个位置,所以如果有数出现了(r-l+1)/2+1次,右边一部分的第一个位置一定是它
综上所述,直接查区间询第(r-l+1)/2+1小,如果它的出现次数>=(r-l+1)/2+1,那它就符合条件,否则不符合条件
#include<cstdio> #include<algorithm> #define N 500001 using namespace std; int n,m,tot,a[N],hash[N]; int root[N],sum[N*20],lc[N*20],rc[N*20],cnt; int ans; void discrete() { sort(a+1,a+n+1); tot=unique(a+1,a+n+1)-(a+1); for(int i=1;i<=n;i++) hash[i]=lower_bound(a+1,a+tot+1,hash[i])-a; } inline void insert(int pre,int & now,int l,int r,int k) { sum[now=++cnt]=sum[pre]+1; if(l==r) return; int mid=l+r>>1; if(k<=mid) { rc[now]=rc[pre]; insert(lc[pre],lc[now],l,mid,k); } else { lc[now]=lc[pre]; insert(rc[pre],rc[now],mid+1,r,k); } } inline void query(int x,int y,int l,int r,int k,int p) { if(l==r) { if(sum[y]-sum[x]>=p) ans=a[l]; else ans=0; return; } int mid=l+r>>1,tmp=sum[lc[y]]-sum[lc[x]]; if(k<=tmp) query(lc[x],lc[y],l,mid,k,p); else query(rc[x],rc[y],mid+1,r,k-tmp,p); } int main() { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) scanf("%d",&a[i]),hash[i]=a[i]; discrete(); for(int i=1;i<=n;i++) insert(root[i-1],root[i],1,tot,hash[i]); int s,t,k; for(int i=1;i<=m;i++) { scanf("%d%d",&s,&t); k=(t-s+1>>1)+1; query(root[s-1],root[t],1,tot,k,k); printf("%d\n",ans); } }