51nod1276岛屿的数量

http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1276&judgeId=597106

 

1276 岛屿的数量 

题目来源： Codility

基准时间限制：1 秒 空间限制：131072 KB 分值: 20 难度：3级算法题

 收藏

 关注

有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

真的佩服自己...........一个位置搞错几个小时没看出来.....

#include<iostream>
#include<string.h>
#include<algorithm>
#define maxn 50005
#define clear(a,b) memset(a,b,sizeof(a))
using namespace std;
struct node
{
    int pos;
    int h;
}is[maxn],que[maxn];
bool flag[maxn];
int ans[maxn];
bool cmp(node a,node b)
{
    return a.h<b.h;
}
int main()
{
        int n,q;
        cin>>n>>q;
        for(int i=0;i<n;i++)
        {
            cin>>is[i].h;
            is[i].pos=i;
        }
        for(int i=0;i<q;i++)
        {
            cin>>que[i].h;
            que[i].pos=i;
        }
        sort(is,is+n,cmp);
        sort(que,que+q,cmp);
        clear(flag,false);
        int j=0,cnt=1;
        for(int i=0;i<q;i++)
        {
            while(is[j].h<=que[i].h&&j<n)
            {
                if(is[j].pos==0)
                {
                    if(flag[is[j].pos+1])
                    {
                        cnt--;
                    }
                }
                else if(is[j].pos==(n-1))
                {
                    if(flag[is[j].pos-1])
                        {
                            cnt--;
                        }
                }
                else
                {
                    if(flag[is[j].pos-1]&&flag[is[j].pos+1])//两边都被淹了
                    {
                        cnt--;
                    }
                    else if(!flag[is[j].pos-1]&&!flag[is[j].pos+1])//两边都没被淹
                    {
                        cnt++;
                    }
                }
                 flag[is[j++].pos]=true;
            }
            ans[que[i].pos]=cnt;
        }
        for(int i=0;i<q;i++)
            cout<<ans[i]<<endl;
    return 0;
}
/*10 10
//116
//40
//94
//31
//108
//98
//119
//34
//64
//36
//75
//39
//7
//30
//44
//17
//81
//84
//114
//106
//3
//3
//1
//1
//4
//1
//3
//3
//2
//3×/