HDU5200 Trees (離線處理)

bigbigship發表於2015-04-05

題目連結:

http://acm.hdu.edu.cn/showproblem.php?pid=5200


題意:

每次去掉高度小於q的樹,每次輸出剩下的塊數。


分析:

我們對高度從高到低進行排序,對要去掉的高度從低到高進行排序。

因此前面去掉的在後面一定會去掉,因可以離線處理,節省時間,

然後需要開一個輔助空間,記錄這棵樹去沒有去掉。


程式碼如下:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
#include <map>
using namespace std;

const int maxn = 50010;

int vis[maxn],ans[maxn];

struct tree{
    int h,id;
    bool operator <(const struct tree &tmp) const{
        return h > tmp.h;
    }
}p[maxn];

struct query{
    int h,id;
    bool operator <(const struct query &tmp) const{
        return h<tmp.h;
    }
}q[maxn];

inline int S(){
    int ret=0,ok=0;
    char c;
    while((c=getchar()))
    {
        if(c>='0'&&c<='9')
        ret=ret*10+c-'0',ok=1;
        else if(ok)
        return ret;
    }
    return ret;
}

int main(){
    int n,m;
    while(~scanf("%d%d",&n,&m)){
        for(int i=0;i<n;i++){
            p[i].h=S();
            p[i].id=i+1;
        }
        sort(p,p+n);
        for(int i=0;i<m;i++){
            q[i].h=S();
            q[i].id=i+1;
        }
        sort(q,q+m);
        int tot = 0;
        memset(vis,0,sizeof(vis));
        for(int i = 0,j = m-1;j>=0;j--){
            for(;i<n;i++){
                if(p[i].h<=q[j].h)
                    break;
                vis[p[i].id]=1;
                if(!vis[p[i].id-1]&&!vis[p[i].id+1]) tot++;
                else if(vis[p[i].id-1]&&vis[p[i].id+1]) tot--;
            }
            ans[q[j].id]=tot;
        }
        for(int i=0;i<m;i++)
            printf("%d\n",ans[i+1]);
    }
    return 0;
}



相關文章