有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
解题思路:
这道题挺有意思的,现在想想也不难,刚开始没想到正确的思路,感觉很难处理。
用一个标记数组标记所有被淹没的岛屿,对于当前的岛屿,如果它左右两边都没淹没,现在要淹没当前的岛屿,一个岛
就会变成两个岛,左右两边都淹没,那么岛的数量就会减少一个,如果左右两边有一个,岛的数量不会变。
这样思路就很清晰了,对a,b数组排序,先淹没小的岛屿,再淹没大的岛屿。
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=50000+100;
struct node1
{
int x;
int cur;
}a[maxn];
struct node2
{
int x;
int cur;
}b[maxn];
bool vis[maxn];
int ans[maxn];
bool cmp1(node1 u,node1 v)
{
return u.x<v.x;
}
bool cmp2(node2 u,node2 v)
{
return u.x<v.x;
}
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++)
{
scanf("%d",&a[i].x);
a[i].cur=i;
}
for(int i=0;i<m;i++)
{
scanf("%d",&b[i].x);
b[i].cur=i;
}
sort(a,a+n,cmp1);
sort(b,b+n,cmp2);
memset(vis,false,sizeof(vis));
int ans1=1;
int j=0;
for(int i=0;i<m;i++)
{
while((a[j].x<=b[i].x)&&j<n)
{
if(a[j].cur==0)
{
if(vis[1])
ans1--;
}
else if(a[j].cur==n-1)
{
if(vis[n-2])
ans1--;
}
else
{
if(vis[a[j].cur-1]&&vis[a[j].cur+1])
ans1--;
else if(!vis[a[j].cur-1]&&!vis[a[j].cur+1])
ans1++;
}
vis[a[j].cur]=true;
j++;
}
ans[b[i].cur]=ans1;
}
for(int i=0;i<m;i++)
cout<<ans[i]<<endl;
return 0;
}