第一次感觉到了程序之美.
就是下面这个简单的代码.
#include <map>
#include <set>
#include <string>
#include <iostream>
#include <vector>
#include <iterator>
#include <algorithm>
#include <sstream>
int main(int argc, char *argv[])
{
std::map<std::string, std::vector<std::string>> families{
{"Ding", {"red", "green", "blue"}},
{"Wang", {"white"}},
};
std::cout << "add new family:" << std::endl;
std::string line;
while (std::getline(std::cin, line))
{
std::stringstream sst(line);
std::istream_iterator<std::string> ist(sst), iste;
const auto la = *ist;
std::copy(++ist, iste, std::back_inserter(families[la]));
}
std::cout << std::endl;
for (auto &last : families)
{
std::cout << last.first << " -> " << std::ends;
std::for_each(last.second.cbegin(), last.second.cend(), [](const std::string &first) { std::cout << first << " "; });
std::cout << "\t#" << std::endl;
}
return 0;
}
实现逻辑不复杂,就是在原有的map 里面去添加新的内容.输入来自控制台.每行对应一个 pair<string,vector> , 每行第一个元素给 pair.first, 后面的元素给到 pair.second.
看看实现,非常的巧妙.
- 在 getline 之后,通过 stringstream 把 string 转成 stream.
- 通过 copy 操作,用插入迭代器 直接从 stringstream 赋值到 vector. 同时,提取获取 stringstream 第一个参数作为 map 的key, 完成 map 的新元素添加.
- 通过范围 for 遍历出 map 的 pair.
- 再通过 for_each + lambda 直接输出 vector 的内容.
以上程序,感觉最好的地方就是下面这两行:
std::copy(++ist, iste, std::back_inserter(families[la])); // 一行代码实现多个操作
std::for_each(last.second.cbegin(), last.second.cend(), [](const std::string &first) { std::cout << first << " "; }); // lambda 简化操作