程序员面试金典 - 面试题 10.10. 数字流的秩（map/树状数组）

文章目录

1. 题目
2. 解题
- 2.1 map
- 2.2 树状数组

1. 题目

假设你正在读取一串整数。每隔一段时间，你希望能找出数字 x 的秩(小于或等于 x 的值的个数)。

请实现数据结构和算法来支持这些操作，也就是说：

实现 track(int x) 方法，每读入一个数字都会调用该方法；
实现 getRankOfNumber(int x) 方法，返回 小于或等于 x 的值的个数。

示例:
输入:
["StreamRank", "getRankOfNumber", "track", "getRankOfNumber"]
[[], [1], [0], [0]]
输出:
[null,0,null,1]提示：
x <= 50000
track 和 getRankOfNumber 方法的调用次数均不超过 2000 次

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/rank-from-stream-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

类似题目：LeetCode 315. 计算右侧小于当前元素的个数（二叉查找树&二分查找&归并排序逆序数总结）

2.1 map

map 存储自己的个数，写入时间复杂度 O(log⁡n)O(\log n)O(logn)
读取秩的时候，从前往后遍历加起来（小于等于x的）O(n)O(n)O(n) 时间复杂度

class StreamRank {map<int,int> m;int count = 0;
public:StreamRank() {}void track(int x) {m[x]++;}int getRankOfNumber(int x) {count = 0;for(auto& mi : m){if(x >= mi.first)count += mi.second;elsebreak;}return count;}
};

108 ms 13.9 MB

map 存储前面小于等于它的个数，读取秩的时间复杂度 O(log⁡n)O(\log n)O(logn)
插入数以后，需要更新所有的 map 的 value，时间复杂度 O(n)O(n)O(n)

class StreamRank {map<int,int> m;
public:StreamRank() {}void track(int x) {auto it = m.rbegin();for(; it != m.rend(); ++it){if(it->first > x)it->second++;//有比x大的，他们的value（比它小的个数） +1elsebreak;}if(it == m.rend() || (it != m.rend() && it->first == x))m[x]++; // map遍历到头了，x不存在，或者x存在elsem[x] = it->second + 1;//遍历没到头，x不存在，x 的 value = 前一个value + 自己}int getRankOfNumber(int x) {if(m.empty() || x < m.begin()->first)return 0;if(m.count(x))return m[x];auto end = m.upper_bound(x);end--;return end->second;}
};

120 ms 14 MB

2.2 树状数组

上面解法：在 n 次操作下的时间复杂度为 O(n2)O(n^2)O(n2)

如何优化：请看树状数组，一次查询和修改时间复杂度均为 O(log⁡n)O(\log n)O(logn)

class StreamRank {vector<int> v;int N = 50002;
public:StreamRank() {v = vector<int>(N);}void track(int x) {update(x+1, 1);}int getRankOfNumber(int x) {return query(x+1);}//-----树状数组-------int lowbit(int x){return x&(-x);}void update(int i, int delta){for( ; i < N; i += lowbit(i))v[i] += delta;}int query(int i){int sum = 0;for( ; i > 0; i -= lowbit(i))sum += v[i];return sum;}
};

44 ms 20.6 MB

程序员面试金典 - 面试题 10.10. 数字流的秩（map/树状数组）相关推荐

程序员面试金典 - 面试题 16.10. 生存人数（自定义优先队列）
文章目录 1. 题目 2. 解题 2.1 优先队列 2.2 双指针 1. 题目给定N个人的出生年份和死亡年份,第i个人的出生年份为 birth[i],死亡年份为 death[i],实现一个方法以计算 ...
程序员面试金典 - 面试题 08.10. 颜色填充（BFS/DFS）
1. 题目颜色填充.编写函数,实现许多图片编辑软件都支持的"颜色填充"功能. 给定一个屏幕(以二维数组表示,元素为颜色值).一个点和一个新的颜色值,将新颜色值填入这个点的周围区域 ...
程序员面试金典 - 面试题 04.10. 检查子树（双重递归）
1. 题目检查子树.你有两棵非常大的二叉树:T1,有几万个节点:T2,有几万个节点. 设计一个算法,判断 T2 是否为 T1 的子树. 如果 T1 有这么一个节点 n,其子树与 T2 一模一样,则 ...
程序员面试金典 - 面试题 17.17. 多次搜索（Trie树）
文章目录 1. 题目 2. 解题 2.1 暴力超时 2.2 Trie树 1. 题目给定一个较长字符串big和一个包含较短字符串的数组smalls,设计一个方法,根据smalls中的每一个较短字符串, ...
程序员面试金典 - 面试题 17.13. 恢复空格（DP+Trie树）
文章目录 1. 题目 2. 解题 2.1 动态规划 2.2 Trie树 1. 题目哦,不!你不小心把一个长篇文章中的空格.标点都删掉了,并且大写也弄成了小写. 像句子"I reset th ...
程序员面试金典 - 面试题 08.13. 堆箱子（DP）
1. 题目堆箱子.给你一堆n个箱子,箱子宽 wi.深 di.高 hi. 箱子不能翻转,将箱子堆起来时,下面箱子的宽度.高度和深度必须大于上面的箱子. 实现一种方法,搭出最高的一堆箱子.箱堆的高度为每 ...
程序员面试金典 - 面试题 17.08. 马戏团人塔（最长上升子序 DP/二分查找）
文章目录 1. 题目 2. 解题 2.1 超时解 2.2 二分查找 1. 题目有个马戏团正在设计叠罗汉的表演节目,一个人要站在另一人的肩膀上.出于实际和美观的考虑,在上面的人要比下面的人矮一点且轻一 ...
程序员面试金典 - 面试题 10.03. 搜索旋转数组
题目难度: 中等原题链接今天继续更新程序员面试金典系列, 大家在公众号算法精选里回复面试金典就能看到该系列当前连载的所有文章了, 记得关注哦~ 题目描述搜索旋转数组.给定一个排序后的数组 ...
程序员面试金典 - 面试题 10.11. 峰与谷
题目难度: 中等原题链接今天继续更新程序员面试金典系列, 大家在公众号算法精选里回复面试金典就能看到该系列当前连载的所有文章了, 记得关注哦~ 题目描述在一个整数数组中,"峰& ...
程序员面试金典 - 面试题 17.07. 婴儿名字
题目难度: 中等原题链接今天继续更新程序员面试金典系列, 大家在公众号算法精选里回复面试金典就能看到该系列当前连载的所有文章了, 记得关注哦~ 题目描述每年,政府都会公布一万个最常见的婴 ...