n棵树,依次编号为1-n,第i棵树的高度为h[i],当编号为x、y的树满足下面条件时,视其属于同一块
1) x+1=y 或 y+1=x;
2) 存在一个棵未被坎掉的树,编号为
z
,
x
和
z
在同一个块并且
y
和
z
也在同一个块。
现在坎掉一些高度不大于某个值的树,问坎掉之后会形成多少个块。
分析:对于每一个询问,若后一个询问高度大于前一个询问的高度,则后一个询问可以在前一个询问的基础上得到。就是说,前一个询问中已经砍掉了的那部分的树,在后一个询问中必然也是要砍掉的。为避免重复计算,可以用离线处理,将每一个询问,按照高度由低到高排序,然后依次处理。
首先对树按照由低到高排序,再对查询进行排序。然后,对于每一个询问,按高度依次删除树高度小于查询高度的树,每一次删除,查看删除的树的两边位置是否有树,若都没有,则砍掉之后会少一个块;若都有,则砍掉后会形成两个块,即增加1个块;若一边有,一边无,则块个数不变。注意对于边界的处理:编号为1的树和编号为n的树的左、右均标记为无树(即已砍倒)。
#include<iostream>
#include<cstdio>
#include<cmath>
#include<map>
#include<cstring>
#include<algorithm>
using namespace std;
#define LL long long
#define N 50005
struct T{
int h,id;
}t[N],p[N];
int ans[N];
bool v[N];
bool cmp(T a,T b) {return a.h<b.h;}
int main()
{
int i,j,n,q;
while(~scanf("%d%d",&n,&q))
{
for(i=0;i<n;i++)
{
v[i+1]=0;
scanf("%d",&t[i].h);
t[i].id=i+1;
}
v[0]=v[n+1]=1;
sort(t,t+n,cmp);
for(i=0;i<q;++i)
{
scanf("%d",&p[i].h);
p[i].id=i;
}
sort(p,p+q,cmp);
int s=1;
for(i=0,j=0;i<q;++i)
{
while(j<n&&t[j].h<=p[i].h)
{
int tem=v[t[j].id+1]+v[t[j].id-1];
v[t[j].id]=1;
if(tem==0) ++s;
else if(tem==2) --s;
j++;
}
ans[p[i].id]=s;
}
for(i=0;i<q;i++)printf("%d\n",ans[i]);
}
return 0;
}