51NOD 1631 小鲨鱼在51nod小学 区间线段树

1631 小鲨鱼在51nod小学
基准时间限制：1 秒 空间限制：131072 KB 分值: 20 难度：3级算法题 收藏  关注
鲨鱼巨巨2.0（以下简称小鲨鱼）以优异的成绩考入了51nod小学。并依靠算法方面的特长，在班里担任了许多职务。

每一个职务都有一个起始时间A和结束时间B，意为小鲨鱼在[A, B]时间内，担任了某职务(inclusively)。

现在给定小鲨鱼的职务履历表，你可以高效的给出小鲨鱼在某天担任了哪些职务吗？

p.s. 由于小鲨鱼担任的职务太多，所有任期小于一个自然月的职务都忽略不计。（如1月1日~2月1日为一个自然月，即月份加1）

p.p.s. 输入数据保证小鲨鱼同时不担任超过200种职务。（牛！）

p.p.p.s 输入的日期均为合法日期，范围在2000年01月01日~2999年12月31日。

p.p.p.p.s巨大的输入输出，推荐使用scanf/printf，编译器推荐使用Virtual C++
Input
第一行为一个整数n，代表小鲨鱼担任过N种职务。(1 <= n <= 10^5)
接下来的n行，每一行为七个整数，y0, m0, d0, y1, m1, d1, x。意为在<y0, m0, d0>到<y1, m1, d1>时间内，小鲨鱼担任了职务x。(1 <= x <= 10^9)
给定的时间皆合法，且起始日期小于或等于截止日期。职务x是唯一的。

接下来是一个整数q，代表q次查询。(1 <= q <= 10^4)
接下来的q行，每一行为三个整数<y, m, d>，代表查询的日期。时间皆合法。
Output
每一次查询输出一行结果。
首先输出一个整数n，代表此时小鲨鱼担任的职务数。（n可以为0）
接下来是n个整数，代表小鲨鱼担任的职务。职务列表保持升序。
Input示例
4
2000 01 01    2000 01 01    111
2000 01 02    2001 02 02    222
2000 01 28    2000 02 29    333
2000 01 29    2000 02 28    444
4
2000 01 01
2000 01 02
2000 01 28
2000 02 29
Output示例
0
1 222
2 222 333
2 222 333
Wizmann (题目提供者)

区间线段树 直接过

---

#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<queue>
#include<sstream>
//#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>

const int N = 400000 + 50;
int X;
char CH;
inline int read(){
    X=0;
    while(!isdigit(CH=getchar()));
    X=CH-'0';
    while(isdigit(CH=getchar())){
        X=X*10+CH-'0';
    }
    return X;
}

struct Node{
    vector<int>work;
    int l,r;
}rmq[5*N];

vector<int>ans;

inline int dateHash(int y,int m,int d){//不妨假设每个月都有31天 一年有372天
    return (y-2000)*372 + 31*(m-1) + d;
}

void initRmq(int k,int l,int r){
    rmq[k].l=l,rmq[k].r=r;
    if(l!=r){
        int mid = (l+r)/2;
        initRmq(2*k+1,l,mid);
        initRmq(2*k+2,mid+1,r);
    }
}

void add(int k,int l,int r,int s,int e,int x){
    if(s<=l&&r<=e){
        rmq[k].work.push_back(x);
        return;
    }
    if(l!=r){
        int mid = (l+r)/2;
        if(s<=mid){
            add(2*k+1,l,mid,s,e,x);
        }
        if(e>=mid+1){
            add(2*k+2,mid+1,r,s,e,x);
        }
    }
}

void query(int k,int l,int r,int q){
    if(l<=q&&q<=r){
        vector<int>&vec=rmq[k].work;
        ans.insert(ans.end(),vec.begin(),vec.end());
    }
    if(l!=r){
        int mid = (l+r)/2;
        if(q<=mid){
            query(2*k+1,l,mid,q);
        }
        if(q>=mid+1){
            query(2*k+2,mid+1,r,q);
        }
    }
}

int main()
{
    //freopen("/home/lu/Documents/r.txt","r",stdin);
    //freopen("/home/lu/Documents/w.txt","w",stdout);
    initRmq(0,1,N);
    int n=read();
    int ys,ye,ms,me,ds,de,x;
    int s,e;
    while(n--){
        ys=read(),ms=read(),ds=read();
        ye=read(),me=read(),de=read();
        x=read();
        s=dateHash(ys,ms,ds);
        e=dateHash(ye,me,de);
        if(e-s>=31){
            add(0,1,N,s,e,x);
        }
    }
    int q=read();
    while(q--){
        ys=read(),ms=read(),ds=read();
        ans.clear();
        query(0,1,N,dateHash(ys,ms,ds));
        sort(ans.begin(),ans.end());
        printf("%d",ans.size());
        for(int i=0;i<ans.size();++i){
            printf(" %d",ans[i]);
        }
        putchar('\n');
    }
    return 0;
}