Codeforces 1642E 线段树

最新推荐文章于 2022-04-24 20:41:32 发布

SHOHOKUKU

最新推荐文章于 2022-04-24 20:41:32 发布

阅读量679

点赞数

分类专栏：数据结构文章标签：算法

本文链接：https://blog.csdn.net/neweryyy/article/details/123171801

版权

数据结构专栏收录该内容

203 篇文章 2 订阅

订阅专栏

本文解析了Codeforces竞赛题目1642E，介绍了如何利用二分查找和线段树数据结构解决关于区间内匿名人群健康状态的问题。通过维护不确定人群集合和优化查询操作，达到O((q+n)logn)的时间复杂度。

摘要由CSDN通过智能技术生成

题意

传送门 Codeforces 1642E Anonymity Is Important

题解

对于 $t = 0$ 的询问，若 $x = 0$ ，则区间内所有人的状态是确定的，即都没有生病。使用 std::set 维护状态仍不确定的人的集合，通过二分求 $[l, r]$ ，就可以 $O(\log n)$ 将明确状态为没有生病的人从集合中删除。

若存在一个 $x = 1$ 的区间 $[l, r]$ ，满足 $l\leq i\leq r$ 且区间内其余人明确为没有生病的状态，则可以判断第 $i$ 个人生病。

设 $l b, u b$ 分别为第 $i$ 个人左右两侧的第一个仍在集合中的人，那么一定不存在左右界都位于 $[l b + 1, i)$ 或 $[i + 1, u b)$ 的询问，满足 $x = 1$ 。那么应用线段树维护 $x = 1$ 且左边界位于 $[l b + 1, i]$ 的查询中，右边界的最小值即可。总时间复杂度 $O\big((q+n)\log n\big)$

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 2E5 + 5, SZ = 1 << 19, INF = 0x3f3f3f3f;
int N, Q, rec[MAXN];
set<int> ps;
int dat[SZ];

void init() { memset(dat, 0x3f, sizeof(dat)); }

void change(int a, int b, int x, int k = 0, int l = 0, int r = N)
{
    if (b <= l || r <= a)
        return;
    if (a <= l && r <= b)
        dat[k] = x;
    else
    {
        int chl = k * 2 + 1, chr = k * 2 + 2, m = (l + r) / 2;
        change(a, b, x, chl, l, m), change(a, b, x, chr, m, r);
        dat[k] = min(dat[chl], dat[chr]);
    }
}

int ask(int a, int b, int k = 0, int l = 0, int r = N)
{
    if (b <= l || r <= a)
        return INF;
    if (a <= l && r <= b)
        return dat[k];
    int chl = k * 2 + 1, chr = k * 2 + 2, m = (l + r) / 2;
    return min(ask(a, b, chl, l, m), ask(a, b, chr, m, r));
}

int main()
{
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    cin >> N >> Q;
    memset(rec, 0x3f, sizeof(rec));
    for (int i = 0; i < N; ++i)
        ps.insert(i);
    init();
    for (int i = 0; i < Q; ++i)
    {
        int type, l, r, x;
        cin >> type;
        if (type == 0)
        {
            cin >> l >> r >> x;
            --l;
            if (x == 0)
            {
                for (auto it = ps.lower_bound(l); it != ps.end() && *it < r;)
                    it = ps.erase(it);
            }
            else
            {
                if (r < rec[l])
                    rec[l] = r, change(l, l + 1, r);
            }
        }
        else
        {
            cin >> l;
            --l;
            if (!ps.count(l))
                cout << "NO\n";
            else
            {
                int lb = -1, ub = N;
                auto it = ps.lower_bound(l);
                if (it != ps.begin())
                    lb = *--it;
                it = ps.upper_bound(l);
                if (it != ps.end())
                    ub = *it;
                cout << (ask(lb + 1, l + 1) <= ub ? "YES" : "N/A") << '\n';
            }
        }
    }
    return 0;
}