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

如果当前有a个岛屿 判断岛i沉没后有多少个岛只要O(1)
对岛按高度排序 2分在log(n)内可以找到沉没的岛的区间 依次对每个岛进行处理
查询q次 每次处理不超过n个岛
所以复杂度是O(qn+qlog(n)+n*log(n))
如果对q次查询按海平面高度排序 再依次处理 那一共处理的岛不会超过n个
复杂度是O(qlog(q)+nlog(n)+n)

#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<string>
#include<vector>
#include<deque>
#include<queue>
#include<algorithm>
#include<set>
#include<map>
#include<stack>
#include<time.h>
#include<math.h>
#include<list>
#include<cstring>
#include<fstream>
//#include<memory.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define pii pair<int,int>
#define INF 1000000007
#define pll pair<ll,ll>
#define pid pair<int,double>
//#define CHECK_TIME

const int N=50000;
struct Query{
    int h,ans;//查询的海平面高度 结果
}qe[N+5];
struct Index{//Query按海平面排序后的新顺序
    Query*p;
}queryIndex[N+5];

int a[N+5];
pii elem [N+5];

inline bool compH(const Index&a,const Index&b){
    return a.p->h<b.p->h;
}

inline int srink(int i,int n){//让岛i沉没 返回沉没后岛屿数改变的数量
    a[i]=0;
    int k=(a[i-1]>0)+(a[i+1]>0);
    return k-1;
}

void slove(int n,int q){
    sort(elem+1,elem+n+1);//将岛屿按高度排序
    sort(queryIndex+1,queryIndex+q+1,compH);//将查询按海平面排序
    pii*st=elem+1,tem={0,INF};
    for(int i=1;i<=q;++i){//[st,end) 表示需要沉没的岛的区间
        tem.first=queryIndex[i].p->h;
        pii*end=upper_bound(st,elem+n+1,tem);
        queryIndex[i].p->ans=i>1?queryIndex[i-1].p->ans:1;
        for(pii*it=st;it!=end;++it)
            queryIndex[i].p->ans+=srink(it->second,n);
        st=end;
    }
}

int main()
{
    //freopen("/home/lu/文档/r.txt","r",stdin);
    //freopen("/home/lu/文档/w.txt","w",stdout);
    int n,q,h;
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;++i){
        scanf("%d",a+i);
        elem[i].first=a[i];
        elem[i].second=i;
    }
    for(int i=1;i<=q;++i){
        scanf("%d",&qe[i].h);
        queryIndex[i].p=&qe[i];
    }
    slove(n,q);
    for(int i=1;i<=q;++i)
        printf("%d\n",qe[i].ans);
    return 0;
}