C++A Text-Query Program Design(文本查询程序设计)

Posted 2021-06-27 nirvana · rebirth

#include <iostream>
#include <cstdio>
#include <string>
#include <vector>
#include <list>
#include <map>
#include <set>
#include <algorithm>
#include <fstream>
#include <sstream>
#include <iterator>
#include <memory>
#include <thread>
using namespace std;

class QueryResult {
	using line_no = vector<string>::size_type;
	friend ostream& print(ostream&, const QueryResult&);
public:
	QueryResult(string s, shared_ptr<set<line_no>> p, shared_ptr<vector<string>> f) :
		sought(s), lines(p), file(f) {}
private:
	string sought;	// 查询单词
	shared_ptr<set<line_no>> lines;		// 出现的行号
	shared_ptr<vector<string>> file;	// 输入文件
};
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() - 1;	// 当前行号
		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>); // 分配一个新的set
			lines->insert(n);		// 将此行号插入set中
		}
	}
}
QueryResult TextQuery::query(const 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);
}


string make_plural(size_t ctr, const string& word, const string& ending) {
	return (ctr == 1) ? word : word + ending;
}
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) 
		os << "\\t(line " << num + 1 << ") " << qr.file->at(num) << endl;
	return os;
}
void runQueries(ifstream& infile) {
	TextQuery 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("1.txt");
	runQueries(infile);
}