poj 2823 Sliding Window

这题我是用线段树来做的。题意看图就明白了!下面是代码,跑了8000多MS...


#include<stdio.h>

#define MAXN 1000001

int MAX(int a,int b)
{
if(a>b)
return a;
else
return b;
}

int MIN(int a,int b)
{
if(a>b)
return b;
else
return a;
}

struct slove
{
int max;
int min;
int l,r;
}tree[3*MAXN];

int d[MAXN];

void build(int l,int r,int s)
{
tree[s].l=l;
tree[s].r=r;
if(l==r)
{
tree[s].max=d[l];
tree[s].min=d[l];
return;
}
int mid=(l+r)>>1;
build(l,mid,s<<1);
build(mid+1,r,s<<1|1);
tree[s].max=MAX(tree[s<<1].max,tree[s<<1|1].max);
tree[s].min=MIN(tree[s<<1].min,tree[s<<1|1].min);
}

int query_max(int l,int r,int s)
{
if(l==tree[s].l&&r==tree[s].r)
return tree[s].max;
int mid=(tree[s].l+tree[s].r)>>1;
if(r<=mid)
return query_max(l,r,s<<1);
else if(l>mid)
return query_max(l,r,s<<1|1);
else
return MAX(query_max(l,mid,s<<1),query_max(mid+1,r,s<<1|1));
}

int query_min(int l,int r,int s)
{
if(l==tree[s].l&&r==tree[s].r)
return tree[s].min;
int mid=(tree[s].l+tree[s].r)>>1;
if(r<=mid)
return query_min(l,r,s<<1);
else if(l>mid)
return query_min(l,r,s<<1|1);
else
return MIN(query_min(l,mid,s<<1),query_min(mid+1,r,s<<1|1));
}

int main()
{
int n,m,i;
while(scanf("%d%d",&n,&m)==2)
{
for(i=1;i<=n;i++)
scanf("%d",&d[i]);
build(1,n,1);
for(i=1;i<n-m+1;i++)
printf("%d ",query_min(i,i+m-1,1));
printf("%d\n",query_min(i,i+m-1,1));
for(i=1;i<n-m+1;i++)
printf("%d ",query_max(i,i+m-1,1));
printf("%d\n",query_max(i,i+m-1,1));
}
return 0;
}
阅读更多
文章标签: tree query build struct
个人分类: poj
上一篇poj 3264 Balanced Lineup
下一篇poj 2845 01000001
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭