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

5 4
2
1
3
2
3
0
1
3
2

Output示例

1
2
0
2

51Nod-1276-岛屿的数量

思路： 当海水淹没一个岛 d，若d相邻的两个岛都被淹没了，则岛屿数量 ans--;

若d相邻的两个岛都没被淹没，则 ans++; 其他情况ans不变。 ***需对边界上的岛特殊分析***

又岛屿数量只与 岛d及其相邻的岛有关，故可对 岛按由低到高排序，同时记录其编号；

对于查询高度，也可按低到高排序，同时记录其编号。

#include<iostream>
#include<algorithm> 
using namespace std;

const int MAX_N=50005;
const int MAX_Q=50005;
struct node{
	int h;
	int vis;
	bool operator<(const node&x){
		return h<x.h;
	}
};
node a[MAX_N],q[MAX_Q];
int n,Q;
int b[MAX_Q];
bool book[MAX_N];

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin>>n>>Q;
	for(int i=0;i<n;++i)
	{
		cin>>a[i].h;
		a[i].vis=i;
	}
	sort(a,a+n);
	for(int i=0;i<Q;++i)
	{
		cin>>q[i].h;
		q[i].vis=i;
	}
	sort(q,q+Q);
	int ans=1;
	for(int j=0,i=0;i<Q;++i)
	{
		int h=q[i].h;
		while(a[j].h<=h&&j<n){
			int d=a[j].vis;
			if(a[j].vis==0){	//内嵌的if 不可合并 
				if(book[1]==true)	ans--;
			}else if(a[j].vis==n-1){
				if(book[n-2]==true)	ans--;
			}else if(book[d-1]==true&&book[d+1]==true)	ans--;
			else if(book[d-1]==false&&book[d+1]==false)	ans++;
			book[a[j].vis]=true;
			j++;
		}
		b[q[i].vis]=ans;
	}
	for(int i=0;i<Q;++i)
		cout<<b[i]<<endl;
	
	return 0;
}