51nod 1276 岛屿的数量(锻炼思维的好题)

题目： http://www.51nod.com/onlineJudge/questionCode.html#problemId=1276¬iceId=21617

有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;
}