码迷,mamicode.com
首页 > 其他好文 > 详细

HDU 4325-Flowers(线段树+离散化)

时间:2015-08-16 18:07:17      阅读:129      评论:0      收藏:0      [点我收藏+]

标签:

题意:

给出每个花开花的时间段,每询问一个时间点输出该时间点开花的数量

分析:

线段树的区间更新,单点查询,但发现时间很大,没法存区间,就想到了离散化。

离散化就是把要处理的数据统一起来重新标号。

#include <map>
#include <set>
#include <list>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <string>
#include <cctype>
#include <complex>
#include <cassert>
#include <utility>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
#define lson l,m,rt<<1
#define pi acos(-1.0)
#define rson m+1,r,rt<<1|1
#define All 1,N,1
#define N 100010
#define read freopen("in.txt", "r", stdin)
const ll  INFll = 0x3f3f3f3f3f3f3f3fLL;
const int INF= 0x7ffffff;
const int mod =  1000000007;
int res[N*8],q[N],sl[N],er[N];
ll sum[N*8],add[N*8],total;
void pushup(int rt){
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void pushdown(int rt,int len){
    if(add[rt]){
        add[rt<<1]+=add[rt];
        add[rt<<1|1]+=add[rt];
        sum[rt<<1]+=1LL*(len-len/2)*add[rt];
        sum[rt<<1|1]+=1LL*(len/2)*add[rt];
        add[rt]=0;
    }
}
void build(int l,int r,int rt){
    sum[rt]=add[rt]=0;
    if(l==r)return;
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    //pushup(rt);
}
void update(int L,int R,int v,int l,int r,int rt){
    if(L<=l&&R>=r){
        add[rt]+=v;
        sum[rt]+=(r-l+1)*v;
        return;
    }
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    if(L<=m)update(L,R,v,lson);
    if(R>m)update(L,R,v,rson);
    pushup(rt);
}
ll query(int v,int l,int r,int rt){
        if(l==r){
            return sum[rt];
        }
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    if(v<=m)return query(v,lson);
    else return query(v,rson);
}
int main()
{
    int t,n,m,test=0;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        int len=0;
        for(int i=0;i<n;++i){
            scanf("%d%d",&sl[i],&er[i]);
            res[++len]=sl[i];
            res[++len]=er[i];
        }
        for(int i=0;i<m;++i){
            scanf("%d",&q[i]);
            res[++len]=q[i];
        }
        sort(res+1,res+len+1);
        int tmp=unique(res+1,res+len+1)-(res+1);
        for(int i=tmp;i>=2;i--)
        {
            if(res[i]!=res[i-1]+1)res[++tmp]=res[i-1]+1;
        }
        sort(res+1,res+tmp+1);
        build(1,tmp,1);
        for(int i=0;i<n;++i){
            int ll=lower_bound(res+1,res+tmp+1,sl[i])-res;
            int rr=lower_bound(res+1,res+tmp+1,er[i])-res;
            update(ll,rr,1,1,tmp,1);
        }
        printf("Case #%d:\n",++test);
       for(int i=0;i<m;++i){
            int p=lower_bound(res+1,res+tmp+1,q[i])-res;
            printf("%I64d\n",query(p,1,tmp,1));
        }
    }
return 0;
}

 

HDU 4325-Flowers(线段树+离散化)

标签:

原文地址:http://www.cnblogs.com/zsf123/p/4734595.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!