test(leetcode): add test for problem #3397

.clang-format

@@ -7,7 +7,6 @@ AllowShortBlocksOnASingleLine: Always
 AlwaysBreakTemplateDeclarations: 'Yes'
 BasedOnStyle: LLVM
 ColumnLimit: 120
-IfMacros: [ONLY, ONCE, TWICE, EVERY]
 IndentWidth: 4
 InsertNewlineAtEOF: true
 SpaceBeforeParens: ControlStatementsExceptControlMacros

leetcode/0001/solution.hpp

@@ -1,3 +1,4 @@
+#pragma once
 #include <cstdlib>
 #include <unordered_map>
 #include <vector>

leetcode/0001/test.cpp

@@ -1,5 +1,5 @@
-#include "test.hpp"
 #include "solution.hpp"
+#include "test_utility.hpp"
 
 using namespace testing;
 namespace {

leetcode/0392/solution.hpp

@@ -1,3 +1,4 @@
+#pragma once
 #include <string>
 
 // 方法1：双指针

leetcode/0392/test.cpp

@@ -1,5 +1,5 @@
-#include "test.hpp"
 #include "solution.hpp"
+#include "test_utility.hpp"
 
 using namespace testing;
 namespace {

leetcode/2011/solution.hpp

@@ -1,3 +1,4 @@
+#pragma once
 #include <algorithm>
 #include <string>
 #include <vector>

leetcode/2011/test.cpp

@@ -1,5 +1,5 @@
-#include "test.hpp"
 #include "solution.hpp"
+#include "test_utility.hpp"
 
 using namespace testing;
 namespace {

leetcode/3346/solution.hpp

@@ -1,3 +1,4 @@
+#pragma once
 #include <algorithm>
 #include <ranges>
 #include <vector>

leetcode/3346/test.cpp

@@ -1,5 +1,5 @@
-#include "test.hpp"
 #include "solution.hpp"
+#include "test_utility.hpp"
 
 using namespace testing;
 namespace {

leetcode/3347/solution.hpp

@@ -1,3 +1,4 @@
+#pragma once
 #include <algorithm>
 #include <ranges>
 #include <vector>

leetcode/3347/test.cpp

@@ -1,5 +1,5 @@
-#include "test.hpp"
 #include "solution.hpp"
+#include "test_utility.hpp"
 
 using namespace testing;
 namespace {

leetcode/3397/solution.hpp

@@ -0,0 +1,20 @@
+#pragma once
+#include <algorithm>
+#include <ranges>
+#include <vector>
+
+// 方法1：贪心
+struct Solution {
+    static int maxDistinctElements(std::vector<int> nums, int k) {
+        std::ranges::sort(nums);
+        int res{1}, prev{nums[0] - k};
+        for (std::size_t i{1}; i < nums.size(); ++i) {
+            if (nums[i] + k > prev) {
+                // 每个数都取可能的最小值
+                prev = std::max(prev + 1, nums[i] - k);
+                ++res;
+            }
+        }
+        return res;
+    }
+};

leetcode/3397/test.cpp

@@ -0,0 +1,39 @@
+#include "solution.hpp"
+#include "test_utility.hpp"
+
+using namespace testing;
+namespace {
+using SolutionList = TypeList<Solution>;
+struct TestCase {
+    TestCase(const YAML::Node &node) {
+        Decode(node["input"]["nums"], input.nums);
+        Decode(node["input"]["k"], input.k);
+        Decode(node["expected"], expected);
+    }
+    struct {
+        std::vector<int> nums;
+        int k;
+    } input;
+    int expected;
+};
+
+template <typename Solution>
+void RunTest(const std::vector<int> &nums, const int k, const int expected) {
+    // 检查约束条件
+    ASSERT_THAT(nums.size(), AllOf(Ge(1), Le(E<5>))); // 1 <= nums.length <= 10^5
+    ASSERT_THAT(nums, Each(AllOf(Ge(1), Le(E<9>))));  // 1 <= nums[i] <= 10^9
+    ASSERT_THAT(k, AllOf(Ge(1), Le(E<9>)));           // 0 <= k <= 10^9
+
+    // 测试目标函数
+    Solution solution;
+    EXPECT_EQ(solution.maxDistinctElements(nums, k), expected);
+}
+
+template <typename Solution>
+void RunTest(const TestCase &tc) {
+    RunTest<Solution>(tc.input.nums, tc.input.k, tc.expected);
+}
+
+TEST_Y(LeetCode3397, Example1);
+TEST_Y(LeetCode3397, Example2);
+} // namespace

leetcode/3397/test_suite.yaml

@@ -0,0 +1,11 @@
+example1:
+  input:
+    nums: [1,2,2,3,3,4]
+    k: 2
+  expected: 6
+
+example2:
+  input:
+    nums: [4,4,4,4]
+    k: 1
+  expected: 3

leetcode/3397/title.txt

@@ -0,0 +1 @@
+maximum-number-of-distinct-elements-after-operations

leetcode/3461/solution.hpp

@@ -1,3 +1,4 @@
+#pragma once
 #include <string>
 
 struct Solution {

leetcode/3461/test.cpp

@@ -1,5 +1,5 @@
-#include "test.hpp"
 #include "solution.hpp"
+#include "test_utility.hpp"
 
 using namespace testing;
 namespace {

leetcode/CMakeLists.txt

@@ -1,5 +1,5 @@
 # 构建测试对象库
-file(GLOB_RECURSE TEST_SRC "*/test.cpp")
+file(GLOB_RECURSE TEST_SRC "*.cpp")
 add_library(test_leetcode_o OBJECT ${TEST_SRC})
 target_include_directories(test_leetcode_o PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/test)
 target_compile_definitions(test_leetcode_o PRIVATE PROBLEM_PATH=\"${CMAKE_CURRENT_SOURCE_DIR}\")

leetcode/test/test.cpp → leetcode/test/test_utility.cpp

@@ -1,4 +1,4 @@
-#include "test.hpp"
+#include "test_utility.hpp"
 
 #include <cctype>
 #include <filesystem>

leetcode/test/test.hpp → leetcode/test/test_utility.hpp

@@ -1,3 +1,4 @@
+#pragma once
 #include "yaml-cpp/yaml.h"
 #include "gmock/gmock.h"
 #include "gtest/gtest.h"