C++Primer 文本查询程序详解

c++primer 文本查询程序 注释详解
初学者主要根据书中写的注释，有错误的地方希望指出，共同学习。

练习12.30：定义你自己版本的 TextQuery 和 QueryResult 类，并执行12.3.1节中的runQueries 函数。

#include<iostream>
#include<vector>
#include<string>
#include<sstream>
#include<fstream>
#include<map>
#include<set>
using namespace std;

class QueryResult;
class TextQuery {
public:
	TextQuery(ifstream&);
	QueryResult query(const string&)const;//表示string出现的那些行
private:
	shared_ptr<vector<string>>file;//保存输入文件
	map<string, shared_ptr<set<vector<string>::size_type>>>wm;//map将文件中每个单词关联到一个动态分配的set上，而set保存了该单词所出现的行号
};
class QueryResult {
	friend ostream& print(ostream&, const QueryResult&);//友元函数，令其可以访问私有成员
public:
	QueryResult(string s, shared_ptr<set<vector<string>::size_type>>p, shared_ptr<vector<string>>f):sought(s),lines(p),file(f){}//初始化
private:
	string sought;//保存查询单词
	shared_ptr<set<vector<string>::size_type>>lines;//保存单词出现行号的set
	shared_ptr<vector<string>>file;//保存输入文件的vector
};
TextQuery::TextQuery(ifstream& is) :file(new vector<string>) {//读取输入文件并建立单词到行号的映射。初始化器分配一个新的vector来保存输入文件中的文本
	string text;
	while (getline(is, text)) {//使用getline逐行读取输入文件
		file->push_back(text);//把读取的文件保存到vector中(file)。因为file是shared_ptr所以需要使用->来解引用
		int n = file->size()-1;//当前行号;size()-1最后一个元素
		istringstream line(text);//使用istringstream来处理读入文件，分解成每个单词
		string word;
		while (line >> word) {//对行中每个单词
			shared_ptr<set<size_t>>& lines = wm[word];//先搜索wm是否存在输入的单词。不在map中，下标运算符会把word添加到wm中，并将值进行值初始化
			if (!lines) //遇到新单词的时候，这个指针是空的，才执行
				lines.reset(new set<vector<string>::size_type>);//分配一个新的set
			lines->insert(n);//将当前行号添加到set中，也会将元素添加到wm中的set中
			//因为lines是引用，所以改变lines也会改变vm
		}
	}
}
QueryResult TextQuery::query(const string& sought)const {
	//定义一个局部static对象，指向空行号set的shared_ptr。如果没有找到给定的单词，返回这个空对象的拷贝
	static shared_ptr<set<vector<string>::size_type>>nodata(new set<vector<string>::size_type>);
	auto loc = wm.find(sought);//使用find而不是下标来查找单词，避免单词添加到wm中
	if (loc == wm.end())
		return QueryResult(sought, nodata, file);//没找到。返回QueryResult的构造函数
	else
		return QueryResult(sought, loc->second, file);
}
ostream& print(ostream& os, const QueryResult& qr) {
	os << qr.sought << " occurs " << qr.lines->size() << " " << ((qr.lines->size()>1) ? "times": "time") << endl;
	for (auto num : *qr.lines)
		os << num + 1 << " " << *(qr.file->begin() + num) << endl;//*(qr.file->begin())是保存的输入文件中的第一行，+num向前移
	return os;
}
void runQueries(ifstream& infile) {
	TextQuery tq(infile);
	while (true) {
		cout << "查找的单词:";
		string s;
		if (!(cin >> s) || s == "q")
			break;
		print(cout, tq.query(s)) << endl;
	}
}
int main()
{
	ifstream word("d:\\a.txt");
	runQueries(word);
}