c++高性能编程hw03

作业地址
个人作业主页

见main.cpp

ppt中很多东西都有，注意使用std::variant时候多结合std::visit和lambda表达式来使用。

#include 
#include 
#include 

// 请修复这个函数的定义：10 分
template 
std::ostream &operator<<(std::ostream &os, std::vector const &a) {
    os << "{";
    for (size_t i = 0; i < a.size(); i++) {
        os << a[i];
        if (i != a.size() - 1)
            os << ", ";
    }
    os << "}";
    return os;
}

// 请修复这个函数的定义：10 分
template 
auto operator+(std::vector const &a, std::vector const &b) {
    // 请实现列表的逐元素加法！10 分
    // 例如 {1, 2} + {3, 4} = {4, 6}
    using T0 = decltype(T1{} + T2{});
    std::vector res;
    
    for (size_t i = 0; i < std::min(a.size(), b.size()); ++i) {
        res.push_back(a[i] + b[i]);
    }
    return res;
}

template 
std::variant operator+(std::variant const &a, std::variant const &b) {
    // 请实现自动匹配容器中具体类型的加法！10 分
   return std::visit(
       [&](const auto& v1, const auto& v2) -> std::variant 
       {
           return v1 + v2;
       },
       a, b); 
}

template 
auto operator+(const std::variant& v1, const std::vector& v2) {
    return v1 + std::variant(v2);
}

template 
std::ostream &operator<<(std::ostream &os, std::variant const &a) {
    // 请实现自动匹配容器中具体类型的打印！10 分
    std::visit(
       [&](const auto& val) {
            os << val;
       }, a);
    return os;
}

int main() {
    std::vector a = {1, 4, 2, 8, 5, 7};
    std::cout << a << std::endl;
    std::vector b = {3.14, 2.718, 0.618};
    std::cout << b << std::endl;
    auto c = a + b;

    // 应该输出 1
    std::cout << std::is_same_v> << std::endl;

    // 应该输出 {4.14, 6.718, 2.618}
    std::cout << c << std::endl;

    std::variant, std::vector> d = c;
    std::variant, std::vector> e = a;
    d = d + c + e;

    // 应该输出 {9.28, 17.436, 7.236}
    std::cout << d << std::endl;

    return 0;
}