C++ Primer 文本查询程序

//QueryResult.h
#include <memory>
#include <vector>
#include <map>
#include <set>
#include <string>

class QueryResult{
	friend std::ostream &print(std::ostream &, const QueryResult &);
public:
	using line_no = std::vector<std::string>::size_type;
	QueryResult(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;		//输入文件
};

//TextQuery.h
#include <memory>
#include <vector>
#include <map>
#include <set>
#include <string>

class QueryResult;		//为了定义query的函数返回类型
class TextQuery{
public:
	using line_no = std::vector<std::string>::size_type;
	TextQuery(std::ifstream&);
	QueryResult 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;
};

//TextQuery.cpp
#include "TextQuery.h"
#include<fstream>	//文件IO
#include<sstream>	//内存IO 即string IO

#include "QueryResult.h"

using namespace std;

//读取输入文件并建立单词到行号的映射
TextQuery::TextQuery(ifstream &is) : file(new vector<string>)
{
	string text;
	while (getline(is, text)) {						//对文件中每一行	
		file->push_back(text);					//保存此行文本
		int n = file->size() - 1;					//当前行号
		//istringstream对象可以绑定一行字符串，然后以空格为分隔符把该行分隔开来
		istringstream line(text);					//将行文本分解为单词
		string word;
		while (line >> word)	{					//对行中每个单词
			//如果单词不在wm中，以之为下标在wm中添加一项
			auto &lines = wm[word];			//lines是一个shared_ptr
			if (!lines)	//在我们第一次遇到这个单词时，此指针为空
				lines.reset(new set<line_no>);
			lines->insert(n);							//将此行号插入set中
		}
	}
}

QueryResult TextQuery::query(const std::string &sought) const
{
	//如果未找到sought，我们将返回一个指向此set的指针
	static shared_ptr<set<line_no>> nodata(new set<line_no>);
	//使用find而不是下标运算符来查找单词，避免将单词添加到wm中！
	auto loc = wm.find(sought);
	if (loc == wm.end())
		return QueryResult(sought, nodata, file);	//未找到
	else
		return QueryResult(sought, loc->second, file);
}

main.cpp

#include <iostream>
#include<fstream>

#include "QueryResult.h"
#include "TextQuery.h"

using namespace std;

string make_plural(size_t ctr, const string &word, const string &ending)
{
	return (ctr == 1) ? word : word + ending;//make_plural(wc, "word ", "s ")当输入中文本中
	//word数大于一是在word后加s，为words为word的复数！
}

ostream &print(ostream &os, const QueryResult &qr)
{
	//如果找打了单词，打印出现的次数和所有出现的位置
	os << qr.sought << " occurs " << qr.lines->size() << " "
		<< make_plural(qr.lines->size(), "time", "s") << endl;
	//打印单词出现的每一行
	for (auto num : *qr.lines)		//对set中每个单词
		//避免行号从0开始给用户带来的困惑
		os << "\t(line " << num + 1 << ")"
		<< *(qr.file->begin() + num) << endl;
	return os;
}

void runQueries(ifstream &infile)
{
	//infile 是一个ifstream, 指向我们要处理的文件
	TextQuery tq(infile);	//保存文件并建立查询map
	//与用户交互：提示用户输入要查询的单词，完成查询并打印结果
	while (true) {
		cout << "enter word to look for, or q to quit: ";
		string s;
		//若遇到文件尾或用户输入了‘q’时循环终止
		if (!(cin >> s) || s == "q") break;
		//指向查询并打印结果
		print(cout, tq.query(s)) << endl;
	}
}

int main()
{
	ifstream in("1.txt");
	runQueries(in);
	system("pause");
	return 0;
}

运行结果：