F - 岛屿的数量

有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行,对应每个查询的岛屿数量。

Sample Input

5 4
2
1
3
2
3
0
1
3
2

Sample Output

1
2
0
2
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=5e4+5;
struct node
{
    int h,pos;
}a[maxn],b[maxn];
bool flag[maxn];//标记岛屿是否被淹没
int ans[maxn];//ans[i]表示海平面为i时岛屿的数量
bool cmp(node x,node y)
{
    return x.h<y.h;
}
int main()
{
    int n,q;
    scanf("%d%d",&n,&q);
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i].h);
        a[i].pos=i;
    }
    for(int i=0;i<q;i++)
    {
        scanf("%d",&b[i].h);
        b[i].pos=i;
    }
    sort(a,a+n,cmp);
    sort(b,b+q,cmp);
    int cnt=1,j=0;
    memset(flag,false,sizeof(flag));
    for(int i=0;i<q;i++)
    {
        while(a[j].h<=b[i].h&&j<n)
        {
            if(a[j].pos==0)//当前岛屿位置为第一个
            {
                if(flag[a[j].pos+1])
                    cnt--;
            }
            else if(a[j].pos==n-1)//当前岛屿的位置为最后一个
            {
                if(flag[a[j].pos-1])
                    cnt--;
            }
            else//在中间
            {
                if(flag[a[j].pos-1]&&flag[a[j].pos+1])//当前山峰的左右两座岛屿都已被淹没
                    cnt--;
                else if(!flag[a[j].pos-1]&&!flag[a[j].pos+1])//左右两座岛都没有淹没
                    cnt++;
                //当只有一侧的岛屿已被淹没,此时岛屿淹没不会影响到岛屿数量
            }
            flag[a[j++].pos]=true;//第j个岛屿被淹没
        }
        ans[b[i].pos]=cnt;
    }
    for(int i=0;i<q;i++)
        printf("%d\n",ans[i]);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_41374539/article/details/82048355
今日推荐