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 <= 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

思路：

很好的一道题，反正又没想出来。

这题的关键要意识到，当一个山峰被水淹没的时候，要么岛屿数+1，要么岛屿数-1，要么不变，当前被淹没的山峰编号为x时，分为三种情况：

1.  如果山峰x-1和山峰x+1都没有被淹没，那么岛屿数就+1。

2.  如果山峰x-1和山峰x+1都已经被淹没，那么岛屿数就-1。

3.  如果山峰x-1和山峰x+1中只有一个被淹没，那么岛屿数不变。

这样我们就可以针对每个一次询问，找到被淹没的山峰，一个一个进行判断，并且不断更新岛屿数res。

然而每次询问不可能扫一遍，所以要提前将询问的山峰和最初的山峰按高度排序，这样每次根据询问更新一段，最后总共就扫了一遍。

另外要注意的一点就是vector保存的是当前被淹没的山峰的编号，这里需要重新排序的，因为如果5，6，7三个都在vector中，这三个需要一起处理，只看4和8。

具体细节看代码。

代码：

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 50005;

struct node {
    int x, id;
    bool operator < (const node &b) const {
        if (x == b.x) return id < b.id;
        return x < b.x;
    }
}a[MAXN], q[MAXN];

int ans[MAXN];
bool vis[MAXN];

int main() {
    //freopen("in.txt", "r", stdin);
    int n, m;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++) {
        scanf("%d", &a[i].x);
        a[i].id = i;
    }
    for (int i = 1; i <= m; i++) {
        scanf("%d", &q[i].x);
        q[i].id = i;
    }
    sort (a + 1, a + 1 + n);
    sort (q + 1, q + 1 + m);
    int pos = 0, res = 1;
    vector <int> vec;
    vis[0] = vis[n + 1] = true;
    for (int i = 1; i <= m; i++) {
        vec.clear();
        while (pos + 1 <= n && a[pos + 1].x <= q[i].x) {
            vec.push_back(a[++pos].id);
            vis[a[pos].id] = true;
        }
        sort (vec.begin(), vec.end());
        int cnt = vec.size(), st = 0, ed = 0;
        while (st < cnt) {
            while (ed + 1 < cnt && vec[ed] + 1 == vec[ed + 1]) ++ed;
            //printf("%d %d\n", vec[st], vec[ed]);
            int tmp = vis[vec[st] - 1] + vis[vec[ed] + 1];
            if (tmp == 2) --res;
            else if (tmp == 0) ++res;
            st = ++ed;
        }
        ans[q[i].id] = res;
    }
    for (int i = 1; i <= m; i++)
        printf("%d\n", ans[i]);
    return 0;
}