在一个下午,我刷到了一个题。里面需要使用一个小根堆,然而里面的需要存的是一个类(或者结构体),堆的排序方式是按照这个类的第三个int类型的成员进行排序,这样当然要重载小于大于符号,于是我就写,竟然不知不觉的搞了半天,这要是在面试的时候基本上可以宣告 “你可以走了”
那我们来看一看我的心路历程:
class Solution {
public:
int kthSmallest(vector<vector<int>>& matrix, int k) {
priority_queue<v> que;
for (int i = 0; i < matrix[0].size(); i++) que.push(v(0, i, matrix[0][i]));
//这大概就是我想要实现的效果,能有一个v的小跟堆
}
};
首先,于是我就写 了一个类,但是类的居然是在这个Slution定义的,我可真牛逼啊,在一个类定义另外一个类
然后,我把v类移到了class Slution外面,但是重载怎么写呢?
其次我又把重载的符号写在了class Slution里面,我可真牛逼啊。肯定又是报错的三,其实这样的错误在IDE环境里面还好,但是放在网络判题器上面根本没法发现;
那么重载应该怎样写呢?
第一种:直接在全局重载这个类的小于符号。
其中最重要的还是要利用greater这个适配器作为介质来处理
class v {
public:
int x;
int y;
int value;
v(int x, int y, int value) :x(x), y(y), value(value) {};
};
bool operator>(const v& v1, const v& v2) {
return v1.value > v2.value;
}
class Solution {
public:
int kthSmallest(vector<vector<int>>& matrix, int k) {
priority_queue<v, vector<v>, greater<v>> que;
for (int i = 0; i < matrix[0].size(); i++) que.push(v(0, i, matrix[0][i]));
//这大概就是我想要实现的效果,能有一个v的小跟堆
}
};
第二种 直接写个仿函数放在que里面;
class v {
public:
int x;
int y;
int value;
v(int x, int y, int value) :x(x), y(y), value(value) {};
};
class mycompare {
public:
bool operator()(v& v1, v& v2) const {
return v1.value > v2.value;
}
};
class Solution {
public:
int kthSmallest(vector<vector<int>>& matrix, int k) {
priority_queue<v, vector<v>, mycompare> que;
for (int i = 0; i < matrix[0].size(); i++) que.push(v(0, i, matrix[0][i]));
//这大概就是我想要实现的效果,能有一个v的小跟堆
}
};
第三种方法 在类中重载 < 符号
class v {
public:
int x;
int y;
int value;
v(int x, int y, int value) :x(x), y(y), value(value) {};
friend bool operator<(const v& v1, const v& v2) {
return v1.value > v2.value;
}
};
class Solution {
public:
int kthSmallest(vector<vector<int>>& matrix, int k) {
priority_queue<v> que;
for (int i = 0; i < matrix[0].size(); i++) que.push(v(0, i, matrix[0][i]));
//这大概就是我想要实现的效果,能有一个v的小跟堆
}
};
注意是< 而不是 > 不然会报错,因为系统默认的是小根堆less<>,如果想重载的是>符号,那么就是que就得写成:
priority_queue<v, vector, greater> que;
其实这种方法和第一种方法是类似的,都是重载了一下<符号
其他方法。是不行的:比如:
1.如果想直接在类中直接重载
那么重载函数必须加const
class v {
public:
int x;
int y;
int value;
v(int x, int y, int value) :x(x), y(y), value(value) {};
bool operator<(const v& v2) const {
return this->value > v2.value;
}
};
2…第三个不能直接传递一个比较函数,而必须是仿函数;
以上都是写个struct/class放在里面,如果有时候想写个多元组,也可以用这种方式
auto tupleCmp = [](const auto& e1, const auto& e2) {
auto&& [x1, y1, d1] = e1;
auto&& [x2, y2, d2] = e2;
return d1 > d2;
};
priority_queue<tuple<int, int, int>, vector<tuple<int, int, int>>, decltype(tupleCmp)> q(tupleCmp);