HDU - 4553 约会安排(线段树)

寒假来了,又到了小明和女神们约会的季节。
  小明虽为屌丝级码农,但非常活跃,女神们常常在小明网上的大段发言后热情回复“呵呵”,所以,小明的最爱就是和女神们约会。与此同时,也有很多基友找他开黑,由于数量实在过于巨大,怎么安排时间便成了小明的一大心事。
  我们已知小明一共有T的空闲时间,期间会有很多女神或者基友来找小明。
  作为一个操作系统曾经怒考71分的大神,小明想到了一个算法,即“首次适应算法”,根据操作系统课本的描述,就是找一段最靠前的符合要求的连续空间分配给每个请求,由此小明做出了一个决定:
  当一个基友来找小明时,小明就根据“首次适应算法”来找一段空闲的时间来和基友约好,如果找到,就说“X,let’s fly”(此处,X为开始时间),否则就说“fly with yourself”;
  当女神来找小明时,先使用一次“首次适应算法”,如果没有找到,小明就冒着木叽叽的风险无视所有屌丝基友的约定,再次使用“无视基友首次适应算法”,两次只要有一次找到,就说“X,don’t put my gezi”(此处,X为开始时间),否则就说“wait for me”
  当然,我们知道小明不是一个节操负无穷的人,如果和女神约会完,还有剩余时间,他还是会和原来约好的基友去dota的。(举个例子:小西(屌丝)和小明约好在15这个时间单位段内打dota,这时候,女神来和小明预约长度为3的时间段,那么最终就是13小明去和女神约会,搞定后在4~5和小西打dota)
  小明偶尔也会想要学习新知识,此时小明就会把某一个时间区间的所有已经预定的时间全部清空用来学习并且怒吼“I am the hope of chinese chengxuyuan!!”,不过小明一般都是三分钟热度,再有人来预定的话,小明就会按耐不住寂寞把学习新知识的时间分配出去。

参考题解
题意:线段树区间合并, 三种命令, D t: 请求一段长度为t的连续区间, 返回区间开始位置; N t: 同上, 但如果没有符合要求, 则忽略之前D的请求再寻找; S t1, t2: 清空[t1, t2];
思路:两个线段树,一个是存女神的,另一个女神屌丝都存,这样找的时候,先找女神屌丝的,如果找不到,就找女神的,直接忽略屌丝。在查询的时候,如果需要的时间比总的时间都多,就直接返回0,否则就先看看左边子树的最大值是不是会比所需大,如果是去左边找,否则看看中间部分,如果可以那么就直接返回中间部分最左侧的位置就可以了,否则去右边找。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1e5+5;
struct node
{
    int lsum, rsum, sum, color;
}ds[maxn<<2], ns[maxn<<2];
char op[10];

void pushUp(node *tree, int cur, int _length)
{
    tree[cur].lsum = tree[cur<<1].lsum;
    tree[cur].rsum = tree[cur<<1|1].rsum;
    if(tree[cur<<1].lsum == _length-(_length>>1))   tree[cur].lsum += tree[cur<<1|1].lsum;
    if(tree[cur<<1|1].rsum == (_length>>1))   tree[cur].rsum += tree[cur<<1].rsum;
    tree[cur].sum = max(tree[cur<<1].sum, max(tree[cur<<1|1].sum, tree[cur<<1].rsum+tree[cur<<1|1].lsum));
}

void pushDown(node *tree, int cur, int _length) //如果是color为1,那么就清空任务
{
    if(tree[cur].color == -1)   return ;
    tree[cur<<1].color = tree[cur<<1|1].color = tree[cur].color;
    tree[cur<<1].lsum = tree[cur<<1].sum = tree[cur<<1].rsum = tree[cur].color ? 0 : _length-(_length>>1);
    tree[cur<<1|1].lsum = tree[cur<<1|1].sum = tree[cur<<1|1].rsum = tree[cur].color ? 0 : (_length>>1);
    tree[cur].color = -1;
}

void build(node *tree, int l, int r, int cur)
{
    tree[cur].color = -1;
    tree[cur].lsum = tree[cur].rsum = tree[cur].sum = r-l+1;
    if(l == r)  return ;
    int mid = (l+r)>>1;
    build(tree, l, mid, cur<<1);
    build(tree, mid+1, r, cur<<1|1);
}

int query(node *tree, int l, int r, int cur, int num)
{
    if(num > tree[cur].sum) return 0;
    if(l == r)  return l;
    pushDown(tree, cur, r-l+1);
    int mid = (l+r)>>1;
    if(tree[cur<<1].sum >= num) return query(tree, l, mid, cur<<1, num);
    if(tree[cur<<1].rsum+tree[cur<<1|1].lsum >= num)    return mid-tree[cur<<1].rsum+1;
    return query(tree, mid+1, r, cur<<1|1, num);
}

void update(node *tree, int L, int R, int l, int r, int cur, int val)
{
    if(L <= l && r <= R)
    {
        tree[cur].color = val;
        tree[cur].lsum = tree[cur].rsum = tree[cur].sum = val ? 0 : (r-l+1);
        return ;
    }
    if(R < l || r < L)  return ;
    pushDown(tree, cur, r-l+1);
    int mid = (l+r)>>1;
    update(tree, L, R, l, mid, cur<<1, val);
    update(tree, L, R, mid+1, r, cur<<1|1, val);
    pushUp(tree, cur, r-l+1);
}

int main()
{
    int T;
    cin >> T;
    for(int cas = 1; cas <= T; cas++)
    {
        int n, t;
        scanf("%d%d", &t, &n);
        printf("Case %d:\n", cas);
        build(ds, 1, t, 1);
        build(ns, 1, t, 1);
        for(int i = 0; i < n; i++)
        {
            int _time, _start, _end;
            scanf("%s", op);
            if(op[0] == 'D')
            {
                scanf("%d", &_time);
                int temp = query(ds, 1, t, 1, _time);
                if(temp)
                {
                    printf("%d,let's fly\n", temp);
                    update(ds, temp, temp+_time-1, 1, t, 1, 1);
                }
                else    printf("fly with yourself\n");
            }
            else if(op[0] == 'N')
            {
                scanf("%d", &_time);
                int temp = query(ds, 1, t, 1, _time);
                if(temp)
                {
                    printf("%d,don't put my gezi\n", temp);
                    update(ds, temp, temp+_time-1, 1, t, 1, temp);
                    update(ns, temp, temp+_time-1, 1, t, 1, temp);
                }
                else
                {
                    temp = query(ns, 1, t, 1, _time);
                    if(temp)
                    {
                        printf("%d,don't put my gezi\n", temp);
                        update(ns, temp, temp+_time-1, 1, t, 1, 1);
                        update(ds, temp, temp+_time-1, 1, t, 1, 1);
                    }
                    else    printf("wait for me\n");
                }
            }
            else
            {
                scanf("%d%d", &_start, &_end);
                update(ds, _start, _end, 1, t, 1, 0);
                update(ns, _start, _end, 1, t, 1, 0);
                printf("I am the hope of chinese chengxuyuan!!\n");
            }
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_40788897/article/details/99439709