diff --git a/main.cpp b/main.cpp index d76cce7..259bf31 100644 --- a/main.cpp +++ b/main.cpp @@ -3,6 +3,7 @@ #include // 请修复这个函数的定义:10 分 +template std::ostream &operator<<(std::ostream &os, std::vector const &a) { os << "{"; for (size_t i = 0; i < a.size(); i++) { @@ -16,19 +17,32 @@ std::ostream &operator<<(std::ostream &os, std::vector const &a) { // 请修复这个函数的定义:10 分 template -std::vector operator+(std::vector const &a, std::vector const &b) { +auto operator+(std::vector const &a, std::vector const &b) { // 请实现列表的逐元素加法!10 分 // 例如 {1, 2} + {3, 4} = {4, 6} + using T0 = decltype(T1{} + T2{}); + std::vector ret; + for(size_t i = 0; i < std::min(a.size(), b.size()); i++) { + ret.push_back(a[i] + b[i]); + } + return ret; } template std::variant operator+(std::variant const &a, std::variant const &b) { // 请实现自动匹配容器中具体类型的加法!10 分 + return std::visit([&] (auto const &t1, auto const &t2) -> std::variant { + return t1 + t2; + }, a, b); } -template -std::ostream &operator<<(std::ostream &os, std::variant const &a) { +template +std::ostream &operator<<(std::ostream &os, std::variant const &a) { // 请实现自动匹配容器中具体类型的打印!10 分 + return std::visit([&] (auto const &t) -> std::ostream& { + os << t; + return os; + }, a); } int main() { @@ -46,7 +60,7 @@ int main() { std::variant, std::vector> d = c; std::variant, std::vector> e = a; - d = d + c + e; + d = d + static_cast(c) + e; // 应该输出 {9.28, 17.436, 7.236} std::cout << d << std::endl;