12章最后一节的程序

#include "pch.h"
#include <iostream>
#include <string>
#include <vector>
#include <map>
#include <set>
#include <sstream>
#include <fstream>
using namespace std;


class QueryResult;
class TextQuery {
public:
	using line_no = vector<string>::size_type;
	TextQuery(ifstream&);
	QueryResult query(const string&) const;
private:
	shared_ptr<vector<string>> file;
	map<string, shared_ptr<set<line_no>>> wm;
};
TextQuery::TextQuery(ifstream &is) : file(new vector<string>) {
	string text;
	while (getline(is, text)) {
		file->push_back(text);
		int n = file->size();//行号
		istringstream line(text);
		string word;
		while (line >> word) {
			auto &lines = wm[word];
			if (!lines)
				lines.reset(new set<line_no>);
			lines->insert(n);
		}
	}
}

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;
	shared_ptr<vector<string>> file;
};
QueryResult TextQuery::query(const string&sought) const {
	static shared_ptr<set< vector<string>::size_type>> nodata(new set< vector<string>::size_type>);
	auto loc = wm.find(sought);
	if (loc == wm.end())
		return QueryResult(sought, nodata, file);
	else
		return QueryResult(sought, loc->second, file);
}
ostream &print(ostream & os, const QueryResult &qr) {
	os << qr.sought << " occurs " << qr.lines->size() << "times" << endl;
	for (auto num : *qr.lines)
		os << "\t(line" << num + 1 << ")" << *(qr.file->begin() + num) << endl;
	return os;
}
void runQueries(ifstream &infile) {
	TextQuery tq(infile);
	while (true) {
		cout << "enter word you want to find: ";
		string s;
		if (!(cin >> s) || s == "q")
			break;
		print(cout, tq.query(s)) << endl;
	}
}





int main() {
	ifstream ifs("ss.txt");
	runQueries(ifs);

	return 0;
}