文件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;
}