51nod 1631 线段树

题目大意:

鲨鱼巨巨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++

思路:将日期转换为整形,构造线段树进行查询即可

ps:如果跨年了则不需要有一个自然月。。


#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
#include <iomanip>

using namespace std;
#define maxn 400005
#define MOD 1000000007
#define mem(a , b) memset(a , b , sizeof(a))
#define LL long long
#define INF 1000000000

int arr[maxn];
set<int>ans;
set<int>::iterator it;
struct node
{
    int l , r , val;
    int ll , rr;
    int flag;
    vector<int>v;

}tree[maxn * 3];

int init()
{
    int pos = 1;
    for(int i = 2000 ; i <= 2999 ; i ++)
    {
        int tmp = i;
        for(int j = 1 ; j <= 12 ; j ++ )
        {
            int tmp2 = tmp * 100 + j;
            for(int k = 1 ; k <= 31 ; k ++)
            {
                arr[pos++] = tmp2 * 100 + k;
            }
        }
    }
    return pos;
}

int get(int y , int m , int d)
{
    return (y * 100 + m ) * 100 + d;
}

void build(int root , int l , int r)
{
    tree[root].v.clear();
    tree[root].l = l;
    tree[root].r = r;
    if(l == r)
    {
        tree[root].ll = arr[l];
        tree[root].rr = arr[l];
        tree[root].flag = 0;
        return;
    }
    int mid = (l + r)>>1;
    build(root<<1 , l , mid);
    build(root<<1|1 , mid + 1 , r);
    tree[root].ll = tree[root<<1].ll;
    tree[root].rr = max(tree[root<<1|1].rr , tree[root<<1].rr);
    tree[root].flag = 0;
}

void update(int root , int l , int r , int val)
{
   // cout << tree[root].ll << " " << tree[root].rr << endl;
   // cout << l << " " << r << endl;
    if(l > tree[root].rr || r < tree[root].ll) return;
    if(tree[root].ll >= l && tree[root].rr <= r )
    {
        tree[root].flag = 1;
     //   cout << "YES" << endl;
        tree[root].v.push_back(val);
        return;
    }
   // int mid = (tree[root].l + tree[root].r)>>1;
    update(root<<1  , l , r , val);
    update(root<<1|1 , l , r , val);

}

void query(int root , int val)
{
    if(val > tree[root].rr || val < tree[root].ll) return;
    if(tree[root].flag && tree[root].ll <= val && tree[root].rr >= val)
    {
        int up = tree[root].v.size();
        for(int i = 0 ; i < up ; i ++) ans.insert(tree[root].v[i]);
    }
    if(tree[root].ll == tree[root].rr ) return;
     query(root << 1 | 1  , val);
    query(root << 1  , val);

}

int main()
{
    int n , q;
    int num = init();
    while(scanf("%d" , &n) != EOF)
    {
        build(1 , 1 , num-1);
        int y0 , m0 , d0 , y1 , m1 , d1 , cs;
        int flag = 0;
        for(int i = 0 ; i < n ; i ++)
        {
            flag = 0;
            scanf("%d %d %d %d %d %d %d" , &y0 , &m0 , &d0 , &y1 , &m1 , &d1 , &cs);
            if( y1 > y0) flag = 1;
            if( !flag && ((m1 - m0 < 1) || (( m1 - m0 ==  1) && d1 < d0)) ) continue;
            else
            {
             //   cout << "hello" << endl;
                update(1 , get(y0 , m0 , d0) , get(y1 , m1 , d1) , cs);
            }
        }
        scanf("%d" , &q);

        for(int i = 0 ; i < q ; i ++)
        {
            ans.clear();
            scanf("%d %d %d" , &y0 , &m0 , &d0);
            int tmp = get(y0 , m0 , d0);

            query(1 , tmp);
           // cout << tmp << " " << i << endl;
            int up = ans.size();
            printf("%d" , up);
            for(it = ans.begin() ; it != ans.end() ; it ++)
            {
                printf(" %d" ,*it);
            }
            printf("\n");
        }
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值