diff --git a/main.cpp b/main.cpp index d76cce7..e334a9a 100644 --- a/main.cpp +++ b/main.cpp @@ -2,7 +2,10 @@ #include #include +#include "cpp_type_name.h" + // 请修复这个函数的定义:10 分 +template std::ostream &operator<<(std::ostream &os, std::vector const &a) { os << "{"; for (size_t i = 0; i < a.size(); i++) { @@ -16,19 +19,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 ans; + for (size_t i = 0; i < a.size() && i < b.size(); i ++) { + T0 c = a[i] + b[i]; + ans.emplace_back(c); + } + return ans; } template std::variant operator+(std::variant const &a, std::variant const &b) { // 请实现自动匹配容器中具体类型的加法!10 分 + return std::visit([&](const auto& t1, const auto& t2) -> std::variant { + return t1 + t2; + }, a, b ); } - template std::ostream &operator<<(std::ostream &os, std::variant const &a) { // 请实现自动匹配容器中具体类型的打印!10 分 + std::visit([&](const auto& input) { + os << input; + }, a); + return os; } int main() { @@ -46,7 +62,7 @@ int main() { std::variant, std::vector> d = c; std::variant, std::vector> e = a; - d = d + c + e; + d = d + std::variant, std::vector>{c} + e; // 应该输出 {9.28, 17.436, 7.236} std::cout << d << std::endl;