最高频率的 ID（线段树）

线段树学习

---

一、题目

你需要在一个集合里动态记录 ID 的出现频率。给你两个长度都为 n 的整数数组 nums 和 freq ，nums 中每一个元素表示一个 ID ，对应的 freq 中的元素表示这个 ID 在集合中此次操作后需要增加或者减少的数目。

增加 ID 的数目：如果 freq[i] 是正数，那么 freq[i] 个 ID 为 nums[i] 的元素在第 i 步操作后会添加到集合中。
减少 ID 的数目：如果 freq[i] 是负数，那么 -freq[i] 个 ID 为 nums[i] 的元素在第 i 步操作后会从集合中删除。
请你返回一个长度为 n 的数组 ans ，其中 ans[i] 表示第 i 步操作后出现频率最高的 ID 数目 ，如果在某次操作后集合为空，那么 ans[i] 为 0 。

输入：nums = [2,3,2,1], freq = [3,2,-3,1]
输出：[3,3,2,2]

二、线段树解法

1.创建线段树

代码如下（示例）：该代码是常用线段树创建模板，数组[1,2,3,4,5]如下：

每个节点p的左右子节点的编号分别为2p和2p+1。

private void build(int p, int l, int r){
        tr[p] = new TreeNode(l, r, 0);
        if(l == r) return;
        int m = (l + r) >> 1;
        build(lc(p), l, m);
        build(rc(p), m + 1, r);
    }

2.递归更新每段里出现最频繁的ID

从树最底层开始更新ID出现频率，依次向上，更新每一段里出现最频繁的ID，最后传入到顶层的就是出现最频繁的ID。
代码如下（示例）：

private void update(int p, int x, int k){
        if(tr[p].l == x && tr[p].r == x){
            tr[p].s += k; return;
        }
        int m = (tr[p].l + tr[p].r) >> 1;
        if(x > m) update(rc(p), x, k);
        else update(lc(p), x, k);
        tr[p].s = Math.max(tr[lc(p)].s, tr[rc(p)].s);
    }

3.完整代码

class TreeNode {
    int l, r;
    long s;

    TreeNode(int l, int r, long s) {
        this.l = l;
        this.r = r;
        this.s = s;
    }
}
public class Solution {
    private TreeNode[] tr;
    private int[] nums;
    private int[] freq;
    
    private void build(int p, int l, int r){
        tr[p] = new TreeNode(l, r, 0);
        if(l == r) return;
        int m = (l + r) >> 1;
        build(lc(p), l, m);
        build(rc(p), m + 1, r);
    }

    private void update(int p, int x, int k){
        if(tr[p].l == x && tr[p].r == x){
            tr[p].s += k; return;
        }
        int m = (tr[p].l + tr[p].r) >> 1;
        if(x > m) update(rc(p), x, k);
        else update(lc(p), x, k);
        tr[p].s = Math.max(tr[lc(p)].s, tr[rc(p)].s);
    }
    
    private int lc(int p) {
        return p << 1;
    }

    private int rc(int p) {
        return (p << 1) | 1;
    }

    public long[] mostFrequentIDs(int[] nums, int[] freq) {
        int maxNum = Integer.MIN_VALUE;
        for (int num : nums) {
            maxNum = Math.max(maxNum, num);
        }
        tr = new TreeNode[maxNum * 4 + 10];
        this.nums = nums;
        this.freq = freq;
        
        build(1, 1, maxNum);
        long[] ret = new long[nums.length];
        for(int i = 0, n = nums.length; i < n; i++){
            update(1, nums[i], freq[i]);
            ret[i] = tr[1].s;
        }
        return ret;
    }
}