文章目录
刷题前唠嗑
LeetCode?启动!!!
今天的题目也比较的简单,因为数据量不大,所以什么做法都能过的去
题目:无限集中的最小数字
题目链接:2336. 无限集中的最小数字
题目描述
代码与解题思路
go
type SmallestInfiniteSet struct {
mp map[int]bool
less 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] = false
tmp := this.less
for i := 1; i < 1001; i++ {
if this.mp[i] == true {
this.less = i
break
}
}
return tmp
}
func (this *SmallestInfiniteSet) AddBack(num int) {
this.mp[num] = true
if num < this.less {
this.less = num
}
}
好吧,我承认我的代码确实是有点屎山,具体来说就是开一个 1000 的 map,然后暴力模拟出来,这种做法和 C++ 直接用 set 自动排序,然后往 set 里面插入 1000 条数据然后 pop 和 push 没啥区别。。非常的暴力
很难过,刷了大半年算法了,磕磕碰碰还是只会暴力解题,哭了,但是看到题目说数据量只有 1000,这谁能忍得住呀呜呜
偷看大佬题解
cpp
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++ 确实是最好写算法的语言