51nod 1276

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

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示例

Output示例

1
2
0
2

可以分情况考虑，如果淹没的点左右都没被淹没则总数加一，如果都被淹没则总数减一，否则不变，可以离线加速。从低到高淹没。

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 struct node{int u,id;}P[50005],Q[50005];
 4 bool cmpid(node A,node B){return A.id<B.id;}
 5 bool cmpu(node A,node B){return A.u<B.u;}
 6 int ans[50005];
 7 bool vis[50005];
 8 int main()
 9 {
10     int N,q,i,j,k,sum=1,cur=1;
11     cin>>N>>q;
12     vis[0]=vis[N+1]=1;
13     for(i=1;i<=N;++i)scanf("%d",&P[i].u),P[i].id=i;
14     for(i=1;i<=q;++i)scanf("%d",&Q[i].u),Q[i].id=i;
15     sort(P+1,P+1+N,cmpu);
16     sort(Q+1,Q+1+q,cmpu);
17     for(i=1;i<=q;++i)
18     {
19         while(cur<=N&&P[cur].u<=Q[i].u){
20             vis[P[cur].id]=1;
21             if(!vis[P[cur].id-1]&&!vis[P[cur].id+1]){
22                 sum++;
23             }
24             else if(vis[P[cur].id-1]&&vis[P[cur].id+1]){
25                 sum--;
26             }
27             cur++;
28         }
29         ans[Q[i].id]=sum;
30     }
31     for(i=1;i<=q;++i)printf("%d
",ans[i]);
32     return 0;
33 }