51Nod 1276 岛屿的数量

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
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 ;
        for(int i=0; i<Q; i++)
        {
            while(a[j].val <= q[i].val && j<N)
            {
                //判断是不是第一个点
                if(a[j].pos == 0)
                {
                    if(vis[1])      //被淹没
                        sum--;
                }
                //判断是不是最后一个点
                else if(a[j].pos == N-1)
                {
                    if(vis[N-2])
                        sum--;
                }
                else
                {
                    if(vis[a[j].pos-1] && vis[a[j].pos+1])//山峰
                        sum--;
                    else if(!vis[a[j].pos-1] && !vis[a[j].pos+1])//山谷
                        sum++;
                }
                vis[a[j].pos] = 1;      //标记被淹没
                j++;
            }
            ans[q[i].pos] = sum;
        }
        for(int i=0; i<Q; i++)
            printf("%d
",ans[i]);
    }
    return 0;
}