先把题目附上:
有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 <= Aii <= 10^9)。
第N + 2 - N + Q + 1行,每行一个数,对应查询的海平面高度(1 <= Qii <= 10^9)。
Output
输出共Q行,对应每个查询的岛屿数量。
Sample Input
5 4
2 1 3 2 3
0 1 3 2
Sample Output
1
2
0
2
当我第一次看到这个题的时候,想到了超时哈哈哈。但是细细一想,假如我们把询问做个排序,也就是让水一点点地长高,这样淹没岛屿是不是就有顺序了。然后就想,总不能每次都把山峰的数组遍历一遍吧,那太可怕了。所以我们可以把山峰也排个序(当然得存储原始的位置),然后让水从左往右一点一点地淹没山峰,每个被淹没的山峰标为0。
还有一点,判断岛屿数量增加或者减少:当正在淹没的岛屿原位置两边都不为0,则山峰数+1,如果两边都是0,山峰数-1,其他情况山峰数不变(为什么可以细想一下)。
本题的精华就是对数组排序的同时保留原位置,然后对答案按照原位置输出。
细节代码里说:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 5e4+5;
typedef struct
{
int h;
int pos;
} node1;
typedef struct
{
int vol;
int pos;
} node2;
bool cmp1(node1 a,node1 b)
{
return a.h<b.h;
}
bool cmp2(node2 a,node2 b)
{
return a.vol<b.vol;
}
int main()
{
int n,m;
node1 a[N];
node2 q[N];
int b[N];
int ans[N];
cin>>n>>m;
memset(a,0,sizeof(a));
memset(q,0,sizeof(q));
memset(ans,0,sizeof(ans));
memset(b,0,sizeof(b));
//上边四行可以不写
for(int i = 1;i<= n;i++)
{
cin>>a[i].h;
b[i] = a[i].h;//保留原山峰位置
a[i].pos = i;
}
sort(a+1,a+n+1,cmp1);//注意此处a+1
for(int i = 1;i<= m;i++)
{
cin>>q[i].vol;
q[i].pos = i;
}
sort(q+1,q+m+1,cmp2);
int num = 1;
int j = 1;
int last = 0;
for(int i = 1;i<= m;i++)
{
while(j<= n&&a[j].h<= q[i].vol)
{
if(b[a[j].pos-1]> 0&&b[a[j].pos+1]> 0).
{
num++;
}
if(b[a[j].pos-1] == 0&&b[a[j].pos+1] == 0)
{
num--;
}//这几句判断是关键
b[a[j].pos] = 0;
j++;//最日狗的就是这个地方,我找了两个小时的错!开始的时候把j++写前面了,这种垃圾错误,丢人!
}
ans[q[i].pos] = num;
}
for(int i = 1;i<= m;i++)
cout<<ans[i]<<endl;
return 0;
如果有不明白或者哪里写的不严谨欢迎留言!
继续加油!