现在的位置: 首页 > 综合 > 正文

【主席树】【bzoj 3524】: [Poi2014]Couriers

2017年04月24日 ⁄ 综合 ⁄ 共 3394字 ⁄ 字号 评论关闭

http://www.lydsy.com/JudgeOnline/problem.php?id=3524

第一眼:莫队,看看数据范围,被卡了

主席树是神马,好吃么?

//#define _TEST _TEST
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;
/************************************************
Code By willinglive    Blog:http://willinglive.cf
************************************************/
#define rep(i,l,r) for(int i=l,___t=(r);i<=___t;i++)
#define per(i,r,l) for(int i=r,___t=(l);i>=___t;i--)
#define MS(arr,x) memset(arr,x,sizeof(arr))
#define LL long long
#define INE(i,u,e) for(int i=head[u];~i;i=e[i].next)
inline const int getint()
{
    int r=0,k=1;char c=getchar();
    for(;c<'0'||c>'9';c=getchar())if(c=='-')k=-1;
    for(;c>='0'&&c<='9';c=getchar())r=r*10+c-'0';
    return k*r;
}
/////////////////////////////////////////////////
int n,m;
int root[500010],ls[10000010],rs[10000010],sum[10000010],sz;
/////////////////////////////////////////////////
void update(int p,int &q,int x,int L,int R)
{
	q=++sz; sum[q]=sum[p]+1;
	if(L==R) return;
	ls[q]=ls[p]; rs[q]=rs[p];
	int mid=L+R>>1;
	if(x<=mid) update(ls[p],ls[q],x,L,mid);
	else 	   update(rs[p],rs[q],x,mid+1,R);
}
int query(int L,int R)
{
	int l=1,r=n,mid,m=R-L+1>>1;
	int x=root[L-1],y=root[R];
	while(l<r)
	{
		if(sum[y]-sum[x]<=m) return 0;
		mid=l+r>>1;
		if(sum[ls[y]]-sum[ls[x]]>m) r=mid,x=ls[x],y=ls[y];
		else if(sum[rs[y]]-sum[rs[x]]>m) l=mid+1,x=rs[x],y=rs[y];
		else return 0;
	}
	return l;
}
/////////////////////////////////////////////////
void input()
{
    scanf("%d%d",&n,&m);
    int x;
    rep(i,1,n)
    {
    	x=getint();
    	update(root[i-1],root[i],x,1,n);
    }
}
void solve()
{
    rep(i,1,m)
    {
    	int l=getint(),r=getint();
    	printf("%d\n",query(l,r));
    }
}
/////////////////////////////////////////////////
int main()
{
    #ifndef _TEST
    freopen("std.in","r",stdin); freopen("std.out","w",stdout);
    #endif
    input();
    solve();
    return 0;
}

update 11-16

昨天晚上脑补了一下主席树,重新写了一遍

tm没加读入优化拖了1s多

//#define _TEST _TEST
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;
/************************************************
Code By willinglive    Blog:http://willinglive.cf
************************************************/
#define rep(i,l,r) for(int i=l,___t=(r);i<=___t;i++)
#define per(i,r,l) for(int i=r,___t=(l);i>=___t;i--)
#define MS(arr,x) memset(arr,x,sizeof(arr))
#define LL long long
#define INE(i,u,e) for(int i=head[u];~i;i=e[i].next)
#define LS ls[o],L,mid
#define RS rs[o],mid+1,R
inline const int getint()
{
    int r=0,k=1;char c=getchar();
    for(;c<'0'||c>'9';c=getchar())if(c=='-')k=-1;
    for(;c>='0'&&c<='9';c=getchar())r=r*10+c-'0';
    return k*r;
}
/////////////////////////////////////////////////
int n,m;
int root[500010];
int ls[10000010],rs[10000010],sum[10000010],sz;
/////////////////////////////////////////////////
void insert(int p,int x,int &o,int L,int R)
{
	o=++sz;
	ls[o]=ls[p]; rs[o]=rs[p]; sum[o]=sum[p]+1;
	if(L==R) return;
	int mid=L+R>>1;
	if(x<=mid) insert(ls[p],x,LS);
	else       insert(rs[p],x,RS);
}
int query(int L,int R)
{
	int l=1,r=n,mid,m=R-L+1>>1;
	int ll=root[L-1],rr=root[R];
	while(l<r)
	{
		if(sum[rr]-sum[ll]<=m) return 0;
		mid=l+r>>1;
		if(sum[ls[rr]]-sum[ls[ll]]>m)r=mid,ll=ls[ll],rr=ls[rr];
		else if(sum[rs[rr]]-sum[rs[ll]]>m)l=mid+1,ll=rs[ll],rr=rs[rr];
		else return 0;
	}
	return l;
}
/////////////////////////////////////////////////
void input()
{
    scanf("%d%d",&n,&m);
    int x;
    rep(i,1,n)
    {
    	scanf("%d",&x);
    	insert(root[i-1],x,root[i],1,n);
    }
}
void solve()
{
	int l,r;
    while(m--)
    {
    	scanf("%d%d",&l,&r);
    	printf("%d\n",query(l,r));
    }
}
/////////////////////////////////////////////////
int main()
{
    #ifndef _TEST
    freopen("std.in","r",stdin); freopen("std.out","w",stdout);
    #endif
    input();
    solve();
    return 0;
}

抱歉!评论已关闭.