LC 3479(2100) 线段树二分 水果成篮

题目
题目：
给你两个长度为 n 的整数数组，fruits 和 baskets，其中 fruits[i] 表示第 i 种水果的 数量，baskets[j] 表示第 j 个篮子的 容量。
你需要对 fruits 数组从左到右按照以下规则放置水果：
每种水果必须放入第一个 容量大于等于 该水果数量的 最左侧可用篮子 中。
每个篮子只能装 一种 水果。
如果一种水果 无法放入 任何篮子，它将保持 未放置。
返回所有可能分配完成后，剩余未放置的水果种类的数量。

思路：
我们需要应用一种数据和结构，可以高效地进行区间查询(查找最左边的符合条件的篮子)和更改(一个篮子装完水果后就没用了) ，那么显然是线段树。
直接套用线段树模板即可，注意下标。

代码：

const int MAXN = 1e5+2;
int tree[MAXN * 4];
const int inf = 1e9;
class Solution {
public:
    int numOfUnplacedFruits(vector<int>& fruits, vector<int>& baskets) {
        int n = fruits.size();
        auto build = [&](this auto&& self, int x, int l, int r) {
            if (l == r) {
                tree[x] = baskets[l - 1];
                return;
            }

            int mid = (l + r) >> 1;
            self(x * 2, l, mid);
            self(x * 2 + 1, mid + 1, r);
            tree[x] = max(tree[x * 2], tree[x * 2 + 1]);
        };

        auto change = [&](this auto &&self, int x, int l, int r, int cp, int cv) {
            if (l > cp || r < cp) return;
            if (l == r) {
                tree[x] = cv;
                return;
            }

            int mid = (l + r) >> 1;
            if (cp <= mid) self(x * 2, l, mid, cp, cv);
            else self(x * 2 + 1, mid + 1, r, cp, cv);

            tree[x] = max(tree[x * 2], tree[x * 2 + 1]);
        };

        auto query = [&](this auto&& self, int x, int l, int r,  int qv) {
            if (tree[x] < qv) return -1;
            if (l == r) return l;
            int mid = (l + r) >> 1;
            if (tree[x * 2] >= qv) {
                int q = self(x * 2, l, mid, qv);
                if (q >= 0) return q;
            }
            return self(x * 2 + 1, mid + 1, r, qv);

        };
        build(1, 1, n);
        
        int ans = 0;
        for (int i = 0; i < n; ++i) {
            int k = query(1, 1, n, fruits[i]);
            if (k < 0) {
                ans++;
            }
            else {
                change(1, 1, n, k, -inf);
            }
        }
        return ans;


    }
};