有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
题解:
可以先对岛屿的高度排序,但要保留岛屿的位置编号,然后有q个询问,把询问的海平面高度排序,依次求淹没后形成的岛屿数量。
那么来分析当前淹没岛屿的情况,如果当前岛屿的左右两侧都还没有被淹没的时候,淹没此岛屿后,岛屿数量+1;当左右两侧都已经被淹没了,淹没后岛屿数量-1;当只有一侧的岛屿被淹没的时候,淹没当前岛屿,岛屿数量不变。
代码如下:
#include<iostream> #include<algorithm> #include<cstdio> #include<cstring> using namespace std; const int maxn=5e4+5; struct node { int h,pos; }a[maxn],b[maxn]; bool flag[maxn];//标记岛屿是否被淹没 int ans[maxn];//ans[i]表示海平面为i时岛屿的数量 bool cmp(node x,node y) { return x.h<y.h; } int main() { int n,q; scanf("%d%d",&n,&q); for(int i=0;i<n;i++) { scanf("%d",&a[i].h); a[i].pos=i; } for(int i=0;i<q;i++) { scanf("%d",&b[i].h); b[i].pos=i; } sort(a,a+n,cmp); sort(b,b+q,cmp); int cnt=1,j=0; memset(flag,false,sizeof(flag)); for(int i=0;i<q;i++) { while(a[j].h<=b[i].h&&j<n) { if(a[j].pos==0)//当前岛屿位置为第一个 { if(flag[a[j].pos+1]) cnt--; } else if(a[j].pos==n-1)//当前岛屿的位置为最后一个 { if(flag[a[j].pos-1]) cnt--; } else//在中间 { if(flag[a[j].pos-1]&&flag[a[j].pos+1])//当前山峰的左右两座岛屿都已被淹没 cnt--; else if(!flag[a[j].pos-1]&&!flag[a[j].pos+1])//左右两座岛都没有淹没 cnt++; //当只有一侧的岛屿已被淹没,此时岛屿淹没不会影响到岛屿数量 } flag[a[j++].pos]=true;//第j个岛屿被淹没 } ans[b[i].pos]=cnt; } for(int i=0;i<q;i++) printf("%d ",ans[i]); return 0; }