hw03 done!

main.cpp

@@ -1,8 +1,9 @@
 #include <iostream>
 #include <vector>
 #include <variant>
-
+#include <algorithm>
 // 请修复这个函数的定义：10 分
+template<class T>
 std::ostream &operator<<(std::ostream &os, std::vector<T> const &a) {
     os << "{";
     for (size_t i = 0; i < a.size(); i++) {
@@ -16,19 +17,48 @@ std::ostream &operator<<(std::ostream &os, std::vector<T> const &a) {
 
 // 请修复这个函数的定义：10 分
 template <class T1, class T2>
-std::vector<T0> operator+(std::vector<T1> const &a, std::vector<T2> const &b) {
+auto operator+(std::vector<T1> const &a, std::vector<T2> const &b) {
     // 请实现列表的逐元素加法！10 分
     // 例如 {1, 2} + {3, 4} = {4, 6}
+    using T0 = decltype(T1{} + T2{});
+    std::vector<T0> res;
+    size_t size = std::min({ a.size(),b.size() });
+    res.reserve(size);
+    for(size_t i = 0;i<size;i++){
+        res.push_back(a[i] + b[i]);
+    }
+    return res;
 }
 
 template <class T1, class T2>
 std::variant<T1, T2> operator+(std::variant<T1, T2> const &a, std::variant<T1, T2> const &b) {
     // 请实现自动匹配容器中具体类型的加法！10 分
+    return std::visit([](auto const &a, auto const &b){
+        return std::variant<T1,T2>{a+b};
+    }, a, b);
+}
+
+template <class T1, class T2, class T3, 
+    std::enable_if_t<std::is_same_v<T1, T3> || std::is_same_v<T2, T3>, bool> = true>
+auto operator+(std::variant<T1, T2> const& a, T3 const &b) {
+    // 请实现自动匹配容器中具体类型的加法！10 分
+    return std::variant<T1, T2>{b} + a;
+}
+
+template <class T1, class T2, class T3,
+    std::enable_if_t<std::is_same_v<T1, T3> || std::is_same_v<T2, T3>, bool> = true>
+    auto operator+(T3 const& b, std::variant<T1, T2> const& a) {
+    // 请实现自动匹配容器中具体类型的加法！10 分
+    return std::variant<T1, T2>{b} + a;
 }
 
 template <class T1, class T2>
 std::ostream &operator<<(std::ostream &os, std::variant<T1, T2> const &a) {
     // 请实现自动匹配容器中具体类型的打印！10 分
+    std::visit([&](auto const &a){
+        os<<a;
+    },a);
+    return os;
 }
 
 int main() {
@@ -46,7 +76,7 @@ int main() {
 
     std::variant<std::vector<int>, std::vector<double>> d = c;
     std::variant<std::vector<int>, std::vector<double>> e = a;
-    d = d + c + e;
+    d = c + d + e;
 
     // 应该输出 {9.28, 17.436, 7.236}
     std::cout << d << std::endl;