引入红黑树近似查询

main.cpp

@@ -4,6 +4,7 @@
 #include <random>
 #include <chrono>
 #include <iomanip>
+#include <gtest/gtest.h>
 
 class RedBlackTree : public ling::RedBlackTree<int> {
 protected:
@@ -16,7 +17,7 @@ protected:
     }
 };
 
-#define DATA_SIZE 10000000
+#define DATA_SIZE 100000
 
 //开始一个统计点
 #define StartStating(name) auto name = std::chrono::high_resolution_clock::now()
@@ -27,7 +28,8 @@ auto duration = std::chrono::duration_cast<std::chrono::microseconds>(__ling_end
 std::cout << text << ": " << std::fixed << std::setprecision(6) <<((double)duration.count() / 1000000) << " 秒(" << duration.count() <<" 微秒)" << std::endl;\
 }while(false)
 
-int main() {
+
+TEST(排序, 堆排序测试) {
     std::random_device rd;
     std::mt19937 gen(rd());
 
@@ -45,11 +47,136 @@ int main() {
     EndStating(start, "排序耗时");
 
     for (int i = 1; i < list.size(); i++) {
-        if (list[i] < list[i - 1]) {
-            std::cout << "错误！" << std::endl;
-            return -1;
-        }
+        ASSERT_TRUE(list[i] >= list[i - 1]);
     }
-
-    return 0;
+}
+
+TEST(红黑树, 搜索测试) {
+    RedBlackTree tree;
+    tree.insert(5);
+    tree.insert(10);
+    tree.insert(15);
+    tree.insert(20);
+    tree.insert(25);
+    tree.insert(30);
+    tree.insert(33);
+    tree.insert(35);
+    tree.insert(40);
+    tree.insert(45);
+    tree.insert(50);
+    auto node = tree.iterativeSearch(1);
+    ASSERT_TRUE(node == nullptr);
+    node = tree.iterativeSearch(4);
+    ASSERT_TRUE(node == nullptr);
+    node = tree.iterativeSearch(5);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 5);
+    node = tree.iterativeSearch(6);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 5);
+    node = tree.iterativeSearch(7);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 5);
+    node = tree.iterativeSearch(8);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 5);
+    node = tree.iterativeSearch(9);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 5);
+    node = tree.iterativeSearch(10);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 10);
+    node = tree.iterativeSearch(11);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 10);
+    node = tree.iterativeSearch(12);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 10);
+    node = tree.iterativeSearch(13);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 10);
+    node = tree.iterativeSearch(14);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 10);
+    node = tree.iterativeSearch(15);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 15);
+    node = tree.iterativeSearch(16);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 15);
+    node = tree.iterativeSearch(17);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 15);
+    node = tree.iterativeSearch(18);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 15);
+    node = tree.iterativeSearch(19);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 15);
+    node = tree.iterativeSearch(20);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 20);
+    node = tree.iterativeSearch(21);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 20);
+    node = tree.iterativeSearch(22);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 20);
+    node = tree.iterativeSearch(23);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 20);
+    node = tree.iterativeSearch(24);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 20);
+    node = tree.iterativeSearch(25);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 25);
+    node = tree.iterativeSearch(26);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 25);
+    node = tree.iterativeSearch(27);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 25);
+    node = tree.iterativeSearch(28);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 25);
+    node = tree.iterativeSearch(29);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 25);
+    node = tree.iterativeSearch(30);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 30);
+    node = tree.iterativeSearch(31);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 30);
+    node = tree.iterativeSearch(32);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 30);
+    node = tree.iterativeSearch(33);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 33);
+    node = tree.iterativeSearch(34);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 33);
+    node = tree.iterativeSearch(35);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 35);
+    node = tree.iterativeSearch(44);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 40);
+    node = tree.iterativeSearch(45);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 45);
+    node = tree.iterativeSearch(46);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 45);
+    node = tree.iterativeSearch(49);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 45);
+    node = tree.iterativeSearch(50);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 50);
+    node = tree.iterativeSearch(1000);
+    ASSERT_FALSE(node == nullptr);
+    ASSERT_EQ(node->value, 50);
 }