有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}。
Input
第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)。
Output
输出共Q行,对应每个查询的岛屿数量。
Input示例
5 4 2 1 3 2 3 0 1 3 2
Output示例
1 2 0 2
思路: 当海水淹没一个岛 d,若d相邻的两个岛都被淹没了,则岛屿数量 ans--;
若d相邻的两个岛都没被淹没,则 ans++; 其他情况ans不变。 ***需对边界上的岛特殊分析***
又岛屿数量只与 岛d及其相邻的岛有关,故可对 岛按由低到高排序,同时记录其编号;
对于查询高度,也可按低到高排序,同时记录其编号。
#include<iostream>
#include<algorithm>
using namespace std;
const int MAX_N=50005;
const int MAX_Q=50005;
struct node{
int h;
int vis;
bool operator<(const node&x){
return h<x.h;
}
};
node a[MAX_N],q[MAX_Q];
int n,Q;
int b[MAX_Q];
bool book[MAX_N];
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cin>>n>>Q;
for(int i=0;i<n;++i)
{
cin>>a[i].h;
a[i].vis=i;
}
sort(a,a+n);
for(int i=0;i<Q;++i)
{
cin>>q[i].h;
q[i].vis=i;
}
sort(q,q+Q);
int ans=1;
for(int j=0,i=0;i<Q;++i)
{
int h=q[i].h;
while(a[j].h<=h&&j<n){
int d=a[j].vis;
if(a[j].vis==0){ //内嵌的if 不可合并
if(book[1]==true) ans--;
}else if(a[j].vis==n-1){
if(book[n-2]==true) ans--;
}else if(book[d-1]==true&&book[d+1]==true) ans--;
else if(book[d-1]==false&&book[d+1]==false) ans++;
book[a[j].vis]=true;
j++;
}
b[q[i].vis]=ans;
}
for(int i=0;i<Q;++i)
cout<<b[i]<<endl;
return 0;
}