文章目录
- 刷题前唠嗑
- 题目:无限集中的最小数字
- 题目描述
- 代码与解题思路
- 偷看大佬题解
- 结语
刷题前唠嗑
LeetCode?启动!!!
今天的题目也比较的简单,因为数据量不大,所以什么做法都能过的去
题目:无限集中的最小数字
题目链接:2336. 无限集中的最小数字
题目描述
代码与解题思路
type SmallestInfiniteSet struct {mp map[int]boolless int
}func Constructor() SmallestInfiniteSet {tmp := map[int]bool{}for i := 1; i < 1001; i++ {tmp[i] = true}return SmallestInfiniteSet{mp: tmp,less: 1,}
}func (this *SmallestInfiniteSet) PopSmallest() int {this.mp[this.less] = falsetmp := this.lessfor i := 1; i < 1001; i++ {if this.mp[i] == true {this.less = ibreak}}return tmp
}func (this *SmallestInfiniteSet) AddBack(num int) {this.mp[num] = trueif num < this.less {this.less = num}
}
好吧,我承认我的代码确实是有点屎山,具体来说就是开一个 1000 的 map,然后暴力模拟出来,这种做法和 C++ 直接用 set 自动排序,然后往 set 里面插入 1000 条数据然后 pop 和 push 没啥区别。。非常的暴力
很难过,刷了大半年算法了,磕磕碰碰还是只会暴力解题,哭了,但是看到题目说数据量只有 1000,这谁能忍得住呀呜呜
偷看大佬题解
class SmallestInfiniteSet {
public:vector<bool> vis;priority_queue<int, vector<int>, greater<int>> q;int idx;SmallestInfiniteSet() : idx(1) {vis.resize(1010, false);}int popSmallest() {int ans = -1;if (!q.empty()) {ans = q.top();q.pop();vis[ans] = false;} else {ans = idx++;}return ans;}void addBack(int x) {if (x >= idx || vis[x]) return;if (x == idx - 1) {idx--;} else {q.push(x);vis[x] = true;}}
};
比较操蛋的事情,大佬的题解如果用 go 来实现,那代码量估计是不小,go 可没有给堆,要我手撕一个那我可又要写屎山了,所以就用 C++ 代码冒充一下,孩子的 C++ 功底还是在的(大概,也可能已经忘光了)
主要的思路是这样的,通过一个 bool 数组来记录这个数是否存在,通过一小根堆的优先级对列维护一个小堆,让我们能 log(N) 的获取存在的最小数。
结语
想念 STL 了,C++ 确实是最好写算法的语言