牛客多校第六场 I - Team Rocket (线段树)

题目链接

给出n个区间和m个询问，每次询问给出一个数，规则是如果这个数位于某一个尚未标记的区间内，就标记这个区间。查询要求输出此次标记的区间数目，强制在线。所有查询结束后要求输出每个区间被标记的时刻。

将所有的区间按照L升序排序，然后将每个区间作为线段树的叶子节点，这里线段树维护的是区间内R的最大值。

对于炸掉的铁路，可以直接将R更新为-INF。每次查询的时候，需要先二分出这个点所能影响到的最左的位置k，之后用这个来进行一些剪枝，如果k大于mid或者位置大于当前的R就不进入，这样将单次查询的复杂度压到了O(logn)。

#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;

const int maxn = 200050;
const int INF = 0x3f3f3f3f;
const ll mod = 998244353;

int t, n, m, cnt, k;
int ans[maxn];
ll res, x;
struct node
{
    int l, r;
    int id;
    node(){}
    node(int L, int R, int ID)
    {
		l = L, r = R, id = ID;
	}
	bool operator < (const node &a) const
	{
		return l < a.l;
	}
}e[maxn];
struct seg
{
    int val;
    int id;
}tree[maxn << 2];
bool cmp(node a, node b)
{
    return a.l < b.l;
}

void build(int root, int l, int r)
{
    if(l > r) return;
    if(l == r)
    {
        tree[root].val = e[l].r;
        tree[root].id = e[l].id;
        return;
    }
    int mid = (l + r) >> 1;
    build(root*2, l, mid);
    build(root*2 + 1, mid + 1, r);
    tree[root].val = max(tree[root*2].val, tree[root*2+1].val);
}

void query(int root, int l, int r, int num)
{
    if(l > r || tree[root].val < x) return;
    if(l == r)
    {
        res = (res*tree[root].id) % mod;
        tree[root].val = -INF;
        cnt++;
        ans[e[l].id] = num;
        return;
    }
    int mid = (l + r) >> 1;
    query(root*2, l, mid, num);
    if(k >= mid + 1) query(root*2 + 1, mid + 1, r, num);
    tree[root].val = max(tree[root*2].val, tree[root*2 + 1].val);
}

int main()
{
    scanf("%d", &t);
    int kase = 0;
    while(t--)
    {
        scanf("%d%d", &n, &m);
        for(int i = 1;i <= n;i++)
        {
            scanf("%d%d", &e[i].l, &e[i].r);
            e[i].id = i;
        }
        memset(ans, 0, sizeof(ans));
        sort(e+1, e+n+1);
        build(1, 1, n);
        cnt = 0, res = 0;
        printf("Case #%d:\n", ++kase);
        for(int i = 1;i <= m;i++)
        {
            scanf("%lld", &x);
            x ^= res;
            k = upper_bound(e+1, e+n+1, node(x, 0, 0)) - e - 1;
            cnt = 0, res = 1;
            if(k > 0) query(1, 1, n, i);
            if(cnt == 0) res = 0;
            printf("%d\n", cnt);
        }
        for(int i = 1;i <= n;i++)
        {
            printf("%d", ans[i]);
            if(i == n) puts("");
            else printf(" ");
        }
    }
    return 0;
}