【线段树】个人练习-Leetcode-3161. Block Placement Queries

题目链接:https://leetcode.cn/problems/block-placement-queries/description/

题目大意:在x轴上,有两种操作:

  • 在某个点上插入一个障碍物
  • 查询,在[0, x]区间内,能否放下长度为sz的物品

插入和查询操作交替进行,返回每次查询的结果。

思路:一开始时看见题目写的"anywhere"以为是【区间内每处都能放下物品】,寻思着那不是整个区间都没有障碍物才行吗,好像判断有点简单了。后来才发现只要有一处能放下就行了。

那其实就是查询[0, x]区间内,被障碍物隔开的各个区间中,能存储的最大长度。这种涉及区间查询的,应该用线段树来解决。然而长久不写已经忘了线段树怎么写了...看了很多题解,还是灵神的比较清晰。

更新i处的值为val

cpp 复制代码
	void update(int o, int l, int r, int i, int val) {
        if (l == r) {
            tree[o] = val;
            return;
        }
        int mid = (l + r) / 2;
        if (i <= mid)
            update(o * 2, l, mid, i, val);
        else
            update(o * 2 + 1, mid+1, r, i, val);
        tree[o] = max(tree[o * 2], tree[o*2 + 1]);
    }

返回[0, x]中最大的区间长度

cpp 复制代码
	int query(int o, int l, int r, int x) {
        if (r <= x)
            return tree[o];
        int mid = (l + r) / 2;
        if (x <= mid) 
            return query(o * 2, l, mid, x);
        return max(tree[o * 2], query(o * 2 + 1, mid + 1, r, x)); 
    }

对于[0, x]来说,要么最大长度在[0, pre]prex左边的最大的障碍物位置)之间,要么在[pre, x]之间(因为每个query中x其实也相当于一个暂时的障碍)。

完整代码

cpp 复制代码
class Solution {
public:
    vector<int> tree;

    void update(int o, int l, int r, int i, int val) {
        if (l == r) {
            tree[o] = val;
            return;
        }
        int mid = (l + r) / 2;
        if (i <= mid)
            update(o * 2, l, mid, i, val);
        else
            update(o * 2 + 1, mid+1, r, i, val);
        tree[o] = max(tree[o * 2], tree[o*2 + 1]);
    }


    int query(int o, int l, int r, int x) {
        if (r <= x)
            return tree[o];
        int mid = (l + r) / 2;
        if (x <= mid) 
            return query(o * 2, l, mid, x);
        return max(tree[o * 2], query(o * 2 + 1, mid + 1, r, x)); 
    }


    vector<bool> getResults(vector<vector<int>>& queries) {
        int maxn = 0;
        for (auto q : queries)
            maxn = max(maxn, q[1]);
        maxn++;
        
        set<int> st{0, maxn};
        tree.resize(2 << (32 - __builtin_clz(maxn)));
        vector<bool> res;
        
        for (auto q : queries) {
            int x = q[1];
            auto it = st.lower_bound(x);
            int pre = *std::prev(it);
            if (q[0] == 1) {
                int nxt = *it;
                st.insert(x);
                update(1, 0, maxn, x, x - pre);
                update(1, 0, maxn, nxt, nxt - x);
            }
            else {
                int maxlen = max(query(1, 0, maxn, pre), x - pre);
                res.push_back(maxlen >= q[2]);
            }
        }
        return res;
    }
};
相关推荐
深圳市快瞳科技有限公司8 分钟前
小场景大市场:猫狗识别算法在宠物智能设备中的应用
算法·计算机视觉·宠物
liulilittle31 分钟前
OPENPPP2 —— IP标准校验和算法深度剖析:从原理到SSE2优化实现
网络·c++·网络协议·tcp/ip·算法·ip·通信
superlls3 小时前
(算法 哈希表)【LeetCode 349】两个数组的交集 思路笔记自留
java·数据结构·算法
田里的水稻3 小时前
C++_队列编码实例,从末端添加对象,同时把头部的对象剔除掉,中的队列长度为设置长度NUM_OBJ
java·c++·算法
纪元A梦3 小时前
贪心算法应用:保险理赔调度问题详解
算法·贪心算法
Jayden_Ruan4 小时前
C++逆向输出一个字符串(三)
开发语言·c++·算法
点云SLAM5 小时前
C++ 常见面试题汇总
java·开发语言·c++·算法·面试·内存管理
叙白冲冲5 小时前
哈希算法以及面试答法
算法·面试·哈希算法
YuTaoShao6 小时前
【LeetCode 每日一题】1277. 统计全为 1 的正方形子矩阵
算法·leetcode·矩阵
古译汉书7 小时前
嵌入式铁头山羊stm32-ADC实现定时器触发的注入序列的单通道转换-Day26
开发语言·数据结构·stm32·单片机·嵌入式硬件·算法