30 lines
734 B
C++
30 lines
734 B
C++
#include <gtest/gtest.h>
|
|
#include <solution/1717.h>
|
|
|
|
class MaximumGainTest : public ::testing::Test
|
|
{
|
|
protected:
|
|
void AssertMaximumGain(const std::string &input, int x, int y, int expected)
|
|
{
|
|
char *s = strdup(input.c_str()); // 复制字符串，防止修改原始数据
|
|
int result = maximumGain(s, x, y);
|
|
ASSERT_EQ(result, expected);
|
|
free(s);
|
|
}
|
|
};
|
|
|
|
// 示例 1
|
|
TEST_F(MaximumGainTest, Example1)
|
|
{
|
|
// 输入：s = "cdbcbbaaabab", x = 4, y = 5
|
|
// 输出：19
|
|
AssertMaximumGain("cdbcbbaaabab", 4, 5, 19);
|
|
}
|
|
|
|
// 示例 2
|
|
TEST_F(MaximumGainTest, Example2)
|
|
{
|
|
// 输入：s = "aabbaaxybbaabb", x = 5, y = 4
|
|
// 输出：20
|
|
AssertMaximumGain("aabbaaxybbaabb", 5, 4, 20);
|
|
} |