首页 > 代码库 > HDU 4417 Super Mario

HDU 4417 Super Mario

题意:

有n个数字  m个查询  每个查询询问区间[l,r]内有几个数字小于h


思路:

由于n个数字是不变的  所以可以利用划分树在nlogn复杂度内求出任意区间的第k小值(这里指建树)

每次查询可以二分查找该区间第几小的数小于等于h  那么答案就是二分出的个数

总复杂度  nlogn + m(logn)^2  即  建树复杂度 + m次询问*每次查询logn*二分复杂度logn

n和m都只有10^5  完全可过


代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 100010
#define M 20

int tree[M][N],toleft[M][N],sorted[N];
int n,m,t,T;

void build(int l,int r,int dep)
{
	if(l==r) return ;
	int i,mid=(l+r)>>1;
	int y=sorted[mid],same=mid-l+1,lpos=l,rpos=mid+1;
	for(i=l;i<=r;i++)
	{
		if(tree[dep][i]<y) same--;
	}
	for(i=l;i<=r;i++)
	{
		if(tree[dep][i]<y) tree[dep+1][lpos++]=tree[dep][i];
		else if(tree[dep][i]==y&&same>0)
		{
			tree[dep+1][lpos++]=tree[dep][i];
			same--;
		}
		else tree[dep+1][rpos++]=tree[dep][i];
		toleft[dep][i]=toleft[dep][l-1]+lpos-l;
	}
	build(l,mid,dep+1);
	build(mid+1,r,dep+1);
}

int query(int L,int R,int l,int r,int dep,int k)
{
	if(l==r) return tree[dep][l];
	int mid=(L+R)>>1,amt=toleft[dep][r]-toleft[dep][l-1];
	if(amt>=k)
	{
		int fl=L+toleft[dep][l-1]-toleft[dep][L-1];
		int fr=fl+amt-1;
		return query(L,mid,fl,fr,dep+1,k);
	}
	else
	{
		int fr=r+toleft[dep][R]-toleft[dep][r];
		int fl=fr-(r-l-amt);
		return query(mid+1,R,fl,fr,dep+1,k-amt);
	}
}

int main()
{
    int i,j,l,r,h,fr,ed,md,tmp,ans;
    while(~scanf("%d",&T))
    {
        for(t=1;t<=T;t++)
        {
            printf("Case %d:\n",t);
            scanf("%d%d",&n,&m);
            for(i=1;i<=n;i++)
            {
                scanf("%d",&sorted[i]);
                tree[0][i]=sorted[i];
            }
            sort(sorted+1,sorted+1+n);
            build(1,n,0);
            while(m--)
            {
                scanf("%d%d%d",&l,&r,&h);
                l++; r++;
                fr=1; ed=r-l+1; ans=0;
                while(fr<=ed)
                {
                    md=(fr+ed)>>1;
                    tmp=query(1,n,l,r,0,md);
                    if(tmp<=h)
                    {
                        fr=md+1;
                        ans=md;
                    }
                    else ed=md-1;
                }
                printf("%d\n",ans);
            }
        }
    }
	return 0;
}


HDU 4417 Super Mario