32 lines
831 B
C++
32 lines
831 B
C++
// This file is generated by mkfile.py
|
||
// Date: 2025-10-18
|
||
|
||
#include <solution/3397.h>
|
||
#include <gtest/gtest.h>
|
||
|
||
class MaxDistinctElementsTest : public ::testing::Test {
|
||
protected:
|
||
void AssertMaxDistinctElements(std::vector<int> nums, int k, int expected) {
|
||
int* arr = new int[nums.size()];
|
||
std::copy(nums.begin(), nums.end(), arr);
|
||
|
||
int result = maxDistinctElements(arr, nums.size(), k);
|
||
ASSERT_EQ(result, expected);
|
||
|
||
delete[] arr;
|
||
}
|
||
};
|
||
|
||
// 示例 1
|
||
TEST_F(MaxDistinctElementsTest, Example1) {
|
||
// 输入：nums = [1,2,2,3,3,4], k = 2
|
||
// 输出：6
|
||
AssertMaxDistinctElements({1, 2, 2, 3, 3, 4}, 2, 6);
|
||
}
|
||
|
||
// 示例 2
|
||
TEST_F(MaxDistinctElementsTest, Example2) {
|
||
// 输入：nums = [4,4,4,4], k = 1
|
||
// 输出：3
|
||
AssertMaxDistinctElements({4, 4, 4, 4}, 1, 3);
|
||
} |