C++文本查询程序 定义类管理数据 用智能指针 C++Primer练习12.30

// TextQueryBook，QueryResultBook类的定义.h文件
#ifndef TEXTQUERYBOOK_H
#define TEXTQUERYBOOK_H
#include<memory>
#include<vector>
#include<string>
#include<map>
#include<set>
#include<fstream>
#include<sstream>
class QueryResultBook;
class TextQueryBook
{
public:
using line_no = std::vector<std::string>::size_type;
TextQueryBook(std::ifstream&);
QueryResultBook query(const std::string&) const;
private:
std::shared_ptr<std::vector<std::string>> file; //输入文件
std::map<std::string, std::shared_ptr<std::set<line_no>>> wm;//每个单词到到它所在行号的集合的映射
};

class QueryResultBook
{
friend std::ostream& print(std::ostream&, const QueryResultBook&);
public:
using line_no = std::vector<std::string>::size_type;
QueryResultBook(std::string s,
std::shared_ptr<std::set<line_no>> p,
std::shared_ptr<std::vector<std::string>> f):
sought(s), lines(p), file(f) { }
private:
std::string sought;  //查询单词
std::shared_ptr<std::set<line_no>> lines; //出现的行号
std::shared_ptr<std::vector<std::string>> file; //输入的文件
};
#endif

// TextQueryBook，QueryResultBook类的实现.cpp文件

#include "TextQueryBook.h"

using namespace::std;
TextQueryBook::TextQueryBook(ifstream& infile):file(new vector<string>)    //创建一个file的指针，指向空的vector<string>
{
string str;
while (getline(infile, str))
{
file->push_back(str);
int num = file->size()-1;  //当前行号
istringstream istr(str);
string word;
while (istr >> word)
{
auto &line = wm[word];  //line为行号的set的指针，如果word是第一次出现，则先添加word后绑定，否则直接line和指针绑定
if (!line) //如果指向的是空对象，说明此word还没有创建新的set，要创建新的set，为后续插入行号做准备
line.reset(new set<line_no>);//重新分配对象给line
line->insert(num);              //如果一个单词在一行中出现多次，则只记录一次
}
}
}
QueryResultBook TextQueryBook::query(const string& word) const
{//查询单词，把结果传给QueryResult
auto ret = wm.find(word);
if (ret != wm.end())
return QueryResultBook(word, ret->second, file);
else
{
static shared_ptr<set<line_no>> p(new set<line_no>);
return QueryResultBook(word,p,file);
}
}
string make_plural(size_t ctr,const string &word,const string &ending)
{
return(ctr > 1) ? word + ending : word;
}
ostream  &print(ostream& out, const QueryResultBook& qr)
{
out << qr.sought << " occurs " << qr.lines->size() << make_plural(qr.lines->size(), "times", "s") << endl;
for (auto wq = qr.lines->begin(); wq != qr.lines->end(); ++wq)
{
out << "(line " << *wq + 1 << ")" << *(qr.file->begin() + *wq) << endl;
}
return out;
}
//测试文件.cpp
#include<iostream>
#include "TextQueryBook.h"
using namespace std;

void RunQueries(ifstream &infile)
{
TextQueryBook tq(infile);
while (true)
{
cout << "enter word to look for,or q to quit: ";
string s;
if (!(cin >> s) || s == "q") break;
print(cout, tq.query(s))<< endl;
}
}
int main()
{
ifstream infile("wordtext.txt");
RunQueries(infile);
return 0;
}