HW03
作业描述
见main.cpp
思路
ppt中很多东西都有,注意使用std::variant时候多结合std::visit和lambda表达式来使用。
#include <iostream>
#include <vector>
#include <variant>
// 请修复这个函数的定义: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++) {
os << a[i];
if (i != a.size() - 1)
os << ", ";
}
os << "}";
return os;
}
// 请修复这个函数的定义:10 分
template <class T1, class T2>
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;
for (size_t i = 0; i < std::min(a.size(), b.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(
[&](const auto& v1, const auto& v2) -> std::variant<T1, T2>
{
return v1 + v2;
},
a, b);
}
template <class T1, class T2, class T3>
auto operator+(const std::variant<T1, T2>& v1, const std::vector<T3>& v2) {
return v1 + std::variant<T1, T2>(v2);
}
template <class T1, class T2>
std::ostream &operator<<(std::ostream &os, std::variant<T1, T2> const &a) {
// 请实现自动匹配容器中具体类型的打印!10 分
std::visit(
[&](const auto& val) {
os << val;
}, a);
return os;
}
int main() {
std::vector<int> a = {1, 4, 2, 8, 5, 7};
std::cout << a << std::endl;
std::vector<double> b = {3.14, 2.718, 0.618};
std::cout << b << std::endl;
auto c = a + b;
// 应该输出 1
std::cout << std::is_same_v<decltype(c), std::vector<double>> << std::endl;
// 应该输出 {4.14, 6.718, 2.618}
std::cout << c << std::endl;
std::variant<std::vector<int>, std::vector<double>> d = c;
std::variant<std::vector<int>, std::vector<double>> e = a;
d = d + c + e;
// 应该输出 {9.28, 17.436, 7.236}
std::cout << d << std::endl;
return 0;
}