有N个岛连在一起形成了一个大的岛屿,如果海平面上升超过某些岛的高度时,则这个岛会被淹没。原本的大岛屿则会分为多个小岛屿,如果海平面一直上升,则所有岛都会被淹没在水下。
给出N个岛的高度。然后有Q个查询,每个查询给出一个海平面的高度H,问当海平面高度达到H时,海上共有多少个岛屿。例如:
岛屿的高度为:{2, 1, 3, 2, 3}, 查询为:{0, 1, 3, 2}。
当海面高度为0时,所有的岛形成了1个岛屿。
当海面高度为1时,岛1会被淹没,总共有2个岛屿{2} {3, 2, 3}。
当海面高度为3时,所有岛都会被淹没,总共0个岛屿。
当海面高度为2时,岛0, 1, 3会被淹没,总共有2个岛屿{3} {3}。
输入
第1行:2个数N, Q中间用空格分隔,其中N为岛的数量,Q为查询的数量(1 <= N, Q <= 50000)。
第2 - N + 1行,每行1个数,对应N个岛屿的高度(1 <= A[i] <= 10^9)。
第N + 2 - N + Q + 1行,每行一个数,对应查询的海平面高度(1 <= Q[i] <= 10^9)。
输出
输出共Q行,对应每个查询的岛屿数量。
输入样例
5 4
2
1
3
2
3
0
1
3
2
输出样例
1
2
0
2
模拟过程。为了不超时,排序一下,从小到大查找,去重
#include<bits/stdc++.h>
using namespace std;
#define ll long long
struct node{
int high,id;
}a[50010],b[50010];
bool cmp(node x,node y)
{
return x.high<y.high;
}
int flag[50010],sum[50010];
int main()
{
ios::sync_with_stdio(false);
int n,q,tmp;
cin>>n>>q;
for(int i=1;i<=n;i++)
{
cin>>a[i].high;
a[i].id=i;
}
for(int i=1;i<=q;i++)
{
cin>>b[i].high;
b[i].id=i;
}
sort(a+1,a+n+1,cmp);
sort(b+1,b+q+1,cmp);
int tip=1,ans=1;
for(int i=1;i<=q;i++)
{
while(b[i].high>=a[tip].high&&tip<=n)
{
tmp=a[tip].id;
flag[tmp]=1;
if(tmp==1&&flag[tmp+1]||tmp==n&&flag[tmp-1])
ans--;
else if(tmp!=1&&tmp!=n)
{
if(flag[tmp-1]&&flag[tmp+1])
ans--;
else if(flag[tmp-1]==0&&flag[tmp+1]==0)
ans++;
}
tip++;
}
sum[b[i].id]=ans;
}
for(int i=1;i<=q;i++)
cout<<sum[i]<<endl;
return 0;
}