岛屿的数量 51Nod - 1276

有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 <= Ai <= 10^9)。
第N + 2 - N + Q + 1行，每行一个数，对应查询的海平面高度(1 <= Qi <= 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 <bits/stdc++.h>
using namespace std;
const int MAXN = 5e4+5;
struct node
{
    int pos, val;
};
node a[MAXN], q[MAXN];
bool cmp(node a, node b)
{
    return a.val < b.val;
}
bool vis[MAXN];
int ans[MAXN];
int main()
{
    int N, Q;
    while(~scanf("%d%d",&N,&Q))
    {
        memset(vis, 0, sizeof(vis));
        for(int i=0; i<N; i++)
        {
            scanf("%d",&a[i].val);
            a[i].pos = i;
        }
        for(int i=0; i<Q; i++)
        {
            scanf("%d",&q[i].val);
            q[i].pos = i;
        }
        sort(a, a+N, cmp);
        sort(q, q+Q, cmp);
        int sum = 1, j = 0 ;///此处sum=1作用为当最小的询问高度比最低的山峰低时就为1
        for(int i=0; i<Q; i++)///当小于询问次数时
        {
            while(a[j].val<=q[i].val && j<N)///当有可被淹没的山峰时，要根据它们的位置判断所剩的山峰数目，也就意味着小于a[j].val的山峰已被淹没(标记)
            {
                ///在进行判断时一定要想着既然j进入判断，则它所在的山峰一定会被淹没
                ///判断是不是第一个点
                if(a[j].pos == 0)
                {
                    if(vis[1])///表示在位置2处的山峰被淹没（下标从0开始），若2处被淹没了，1处也会被淹没（既然进来了就表示1处高度小于当前询问高度），数量减少，否则不变
                        sum--;
                }
                ///判断是不是最后一个点
                else if(a[j].pos == N-1)///同位置1处判断一致，如果倒数第二个被淹没（标记）了，最后一个也会被淹没
                {
                    if(vis[N-2])
                        sum--;
                }
                ///中间的点
                else
                {
                    if(vis[a[j].pos-1] && vis[a[j].pos+1])///山峰，如果J处两边都被标记淹没了，周围只有它自己没被淹没，当水的高度可以淹没它是，数量也应该减少一
                        sum--;
                    else if(!vis[a[j].pos-1] && !vis[a[j].pos+1])///山谷，试想若j处被标记淹没了，但是两边都没被标记淹没，相当于把一块木头从中间锯断，是不是增加一
                        sum++;
                    ///else 当J处两边只有一边被淹没时，把它淹没数量不会减少，所以不需要写（注意，j位于中间不是两边）
                }
                vis[a[j].pos] = 1;///标记被淹没了
                j++;
            }
            ans[q[i].pos] = sum;
        }
        for(int i=0; i<Q; i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}