Я попробовал два метода сравнения, но не получил ожидаемого результата. Первый метод сравнения находится внутри настроенного класса MyTreeNode
как operator<
; Второй - это новый класс сравнения compare
с методом переопределения operator()(MyTreeNode*)
.
Код показан ниже. Оба выхода:
5 6 1
, в то время как ожидаемый порядок должен быть 1 5 6. Правило для порядка: Если два узла имеют одинаковый x
, то узел с большим значением y
приходит первым Если узлы имеют одинаковые значения x
и y
, то на первом месте стоит узел с меньшим значением treeNode->val
.
Так, кто-нибудь может помочь мне объяснить это? Спасибо
#include <vector>
#include <cstddef>
#include <algorithm>
#include <iostream>
using namespace std;
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x) {}
};
class MyTreeNode{
public:
TreeNode* treeNode;
int x;
int y;
public:
MyTreeNode(TreeNode* node, int _x, int _y): treeNode(node), x(_x), y(_y){
}
//Solution 1.
bool operator<(MyTreeNode& node){
if(x< node.x){
return true;
}
if(x == node.x && y > node.y){
return true;
}
if(x == node.x && y == node.y
&&treeNode->val<node.treeNode->val){
return true;
}
return false;
}
};
//Solution 2
class compare{
public:
bool operator()(MyTreeNode* node1, MyTreeNode* node2){
if(node1->x < node2->x){
return true;
}
if(node1->x == node2->x && node2->y > node1->y){
return true;
}
if(node1->x == node2->x && node2->y == node1->y
&&node1->treeNode->val<node2->treeNode->val){
return true;
}
return false;
}
};
int main(int argc, char* argv[]){
//Solution so;
vector<MyTreeNode*> trees;
trees.push_back(new MyTreeNode(new TreeNode(5), 0, -2)); //A
trees.push_back(new MyTreeNode(new TreeNode(6), 0, -2)); //B
trees.push_back(new MyTreeNode(new TreeNode(1), 0, 0)); //C
//Solution 1
sort (trees.begin(), trees.end());
//Solution 2
//sort (trees.begin(), trees.end(), compare()); // print 5 6 1
// for(int i=0; i<res.size(); i++){
for_each(trees.begin(), trees.end(), [](const MyTreeNode* ele){cout<< " "<< ele->treeNode->val ;});
//}
}
```