UPCC:2224 Boring Counting(2分+ツリー分割)
各照会区間はA以下B以下の数字より大きい.
考え方:二分+木を分ける.木を分けて区間内の第kの小さい数字を求めることができて、2分に協力してある数字を求めることができます.区間内で何番目の小さい数字ですか?
考え方:二分+木を分ける.木を分けて区間内の第kの小さい数字を求めることができて、2分に協力してある数字を求めることができます.区間内で何番目の小さい数字ですか?
#include <iostream>
#include <vector>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn=50005;
typedef long long LL;
struct DivideTree
{
int arr[maxn],sorted[maxn],dat[20][maxn];
int toleft[20][maxn];
void build(int c,int L,int R)
{
int mid=(L+R)/2,lsame=mid+1-L,lp=L,rp=mid+1;
for(int i=L; i<mid; ++i)
if(sorted[i]<sorted[mid]) lsame--;
for(int i=L; i<=R; ++i)
{
if(i==L)
{
toleft[c][i]=0;
}
else
{
toleft[c][i]=toleft[c][i-1];
}
if(dat[c][i]<sorted[mid])
{
dat[c+1][lp++]=dat[c][i];
toleft[c][i]++;
}
else if(dat[c][i]>sorted[mid])
dat[c+1][rp++]=dat[c][i];
else
{
if(lsame)
{
lsame--;
toleft[c][i]++;
dat[c+1][lp++]=sorted[mid];
}
else
{
dat[c+1][rp++]=sorted[mid];
}
}
}
if(L==R) return ;
build(c+1,L,mid);
build(c+1,mid+1,R);
}
int query(int c,int L,int R,int ql,int qr,int k)
{
if(L==R)
return dat[c][L];
int mid=(L+R)/2;
int la,lb,ra,rb;
if(L==ql)
la=0;
else
la=toleft[c][ql-1];
lb=toleft[c][qr];
ra=ql-L-la;
rb=qr+1-L-lb;
int s=lb-la;
if(k<=s)
return query(c+1,L,mid,L+la,L+lb-1,k);
else
return query(c+1,mid+1,R,mid+1+ra,mid+rb,k-s);
}
};
DivideTree tree;
int n,q;
int binSearch1(const int &ql,const int &qr,const int &target)
{
int low=1,high=qr-ql+1;
int mid=low+(high-low)/2;
while(low<high)
{
if(tree.query(0,1,n,ql,qr,mid)>=target) high=mid;
else low=mid+1;
mid=low+(high-low)/2;
}
if(tree.query(0,1,n,ql,qr,mid)>=target)
return mid;
else
return -1;
}
int binSearch2(const int &ql,const int &qr,const int &target)
{
int low=1,high=qr-ql+1;
int mid=low+(high-low+1)/2;
while(low<high)
{
if(tree.query(0,1,n,ql,qr,mid)<=target) low=mid;
else high=mid-1;
mid=low+(high-low+1)/2;
}
if(tree.query(0,1,n,ql,qr,mid)<=target)
return mid;
else
return -1;
}
int main()
{
int T,kase=0;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&q);
for(int i=1; i<=n; ++i)
{
scanf("%d",&tree.arr[i]);
tree.sorted[i]=tree.dat[0][i]=tree.arr[i];
}
sort(tree.sorted+1,tree.sorted+n+1);
tree.build(0,1,n);
printf("Case #%d:
",++kase);
while(q--)
{
int l,r,a,b;
scanf("%d%d%d%d",&l,&r,&a,&b);
int x,y;
x=binSearch1(l,r,a);
y=binSearch2(l,r,b);
if(x==-1||y==-1) puts("0");
else printf("%d
",y-x+1);
}
}
return 0;
}