C++用多个容器实现文本查询TextQuery程序

文件map_file.txt放到工程下面的文件夹中

下面是TextQuery类的实现

//TextQuery类
//TextQuery.h文件
#include <iostream>
#include <string>
#include <vector>
#include <map>
#include <fstream>
#include <sstream>
#include <set>

using namespace std;

class TextQuery
{
public:
	typedef vector<string>::size_type line_no;
	//read_file函数在类内部定义，是内联函数
	void read_file(ifstream &is)
	{
		store_file(is);
		build_map();
	}
	//成员函数
	set<line_no> run_query(const string&) const;
	string text_line(line_no) const;
private:
	void store_file(ifstream&);
	void build_map();
	//lines_of_text用来保存读取文件中的每一行的单词
	vector<string> lines_of_text;
	//当单词与行号用map容器存放
	map< string , set<line_no> > word_map;
};
//成员函数的实现
//读取要打开的文件，将每一行的单词存入到vector容器中
void TextQuery::store_file(ifstream &is)
{
	string textline;
	//每次存储一行数据
	while(getline(is,textline))
	{
		lines_of_text.push_back(textline);
	}
}
//实现build_map成员函数，此函数实现了建立单词与行号的的map容器
void TextQuery::build_map()
{
	//line_no的类型是vector的size_type
	for(line_no line_num = 0; line_num != lines_of_text.size(); ++line_num)
	{//把vector中每一行的单词用循环的方式全都存入到word_map容器中便于下一步的查询
		//定义一个string的输入流
		istringstream stream(lines_of_text[line_num]);
		string word;
		//每次输入一个单词
		while(stream >> word)
		{//插入map容器时不能用make_pair(word,line_num)
			//插入到word_map的map容器中
			//如果单词存在插入该单词所在的行，如果单词不存在，插入单词同时插入该单词所在的行
			//所以map容器中单词键多对应的值是一个set容器，里面放的是vector容器，存在单词所在的所有行
			word_map[word].insert(line_num);
		}
	}
}
//实现查询功能的成员函数
//参数使用const引用提高效率，不用复制实参
set<TextQuery::line_no> TextQuery::run_query(const string &query_word) const
{//此处查询
	//利用map容器的find函数如果存在返回指向该元素的迭代器，如果不存在则返回超出末端迭代器
	map< string , set<line_no> >::const_iterator loc = word_map.find(query_word);
	if(loc == word_map.end())
		//如果查询的单词不存在返回空的set<line_no>，默认构造函数返回0
		return set<line_no>();
	else
		//如果单词存在，返回该单词所对应的set容器，里面用vector容器存放着所有行数
		return loc->second;
}
//当通过run_query查到该行有此单词时，不仅要返回该行的行号，同时还需要输出出现该单词的每一行
string TextQuery::text_line(line_no line) const
{//输出该单词所在的行，返回string类型
	if(line < lines_of_text.size())
		return lines_of_text[line];
	throw out_of_range("line number out of range");
}

主函数

#include "TextQuery.h"
#include <iostream>
#include <string>
#include <fstream>

using namespace std;

//打开文件用于读的函数
ifstream& open_file(ifstream &in , const string &file)
{
	in.close();
	in.clear();
	in.open(file.c_str());
	return in;
}
//返回单词复数的
string make_plural(size_t cnt , string s1, string s2)
{
	return (cnt == 1)? s1 : s1+s2;
}
//打印结果的函数
void print_results(const set<TextQuery::line_no> &locs , const string &sought , const TextQuery &file)
{
	typedef set<TextQuery::line_no> line_nums;
	line_nums::size_type size = locs.size();
	cout << sought << " occurs " << size << " " << make_plural(size,"time","s") << endl;
	line_nums::const_iterator it = locs.begin();
	//如果所查找的单词在某一行，则输出该行的内容
	for(; it != locs.end() ; ++it)
	{
		//这里+1将输出的第0行变为第一行
		cout << "(line" <<(*it) +1 <<")" << "   " << file.text_line(*it) << endl;
	}
}
int main()
{
	//infile是一个文件输入流的对象
	ifstream infile;
	//判断打开文件是否成功
	if(!open_file(infile,"map_file.txt"))
	{
		throw runtime_error("file can not open");
		return -1;
	}
	TextQuery tq;
	//调用TextQuery类中的read_file成员函数
	tq.read_file(infile);
	while(true)
	{
		cout << "please input word to look for" << endl;
		string s;
		cin >> s;
		if(s == "0")
		{//结束判断
			break;
		}
		//查询输入的单词s
		set<TextQuery::line_no> locs = tq.run_query(s);
		print_results(locs , s , tq);

	}
	return 0;
}