容器的综合应用：文本查询程序

对查询类的编写：

class TextQuery 
{
public:
 TextQuery();
 virtual ~TextQuery();
 typedef std::vector<std::string>::size_type line_no;   //起个别名要么太长让人看着蛋疼

 void read_file(std::ifstream &is)
 {
  store_file(is);
  build_map();
 }
 std::set<line_no> run_query(const std::string&) const;
 std::string text_line(line_no) const;
private:
 void store_file(std::ifstream&);
 void build_map();
 std::vector<std::string> lines_of_text;          //保存输入文件
 std::map<std::string, std::set<line_no> > word_map;        //将出现单词于出现该单词的行号集合相关

};

类成员函数的实现：

//存储输入文件
void TextQuery::store_file(ifstream &is)
{
 string textline;
 while(getline(is, textline))
  lines_of_text.push_back(textline);
}

//建立单词map容器
void TextQuery::build_map()
{
 for (line_no line_num =0;
 line_num!=lines_of_text.size();
 ++line_num)
 {
  istringstream line(lines_of_text[line_num]);

  string word;
  while(line>>word)
   word_map[word].insert(line_num);
 }
}

//实现查询功能
set<TextQuery::line_no>TextQuery::run_query(const string &query_word) const
{
 map<string, set<line_no> >::const_iterator loc=word_map.find(query_word);         //为了避免在word中加入单词，使用find函数，若使用下标将会加入单词

 if(loc==word_map.end())
  return set<line_no>();
 else
  return loc->second;
}

//run_query运行后输出出现单词出现的每一行
string TextQuery::text_line(line_no line) const
{
 if(line < lines_of_text.size())
  return lines_of_text[line];
 throw std::out_of_range("line number out of range");
}

辅助函数的实现：

/*这个函数要么返回形参word的副本，要么返回一个未命名的临时string对象
   这个临时对象是由字符串word与ending的相加产生的，这两种情况下return都在调用
   该函数的地方复制了返回的string对象*/

string make_plural(size_t ctr, const string &word,
       const string &ending)
{
 return (ctr==1)? word: word +ending;
}

ifstream & open_file(ifstream &in, const string &file)
{
 in.close();
 in.clear();
 in.open(file.c_str());
 return in;
}

/*输出结果*/
void print_result(const set<TextQuery::line_no>& locs,
      const string sought,const TextQuery &file)
{
 ofstream outfile(".\\result.txt",ofstream::app);  //在每次打开前找到文件尾，就是上次的数据不被覆盖追加数据

 typedef set<TextQuery::line_no> line_nums;
 line_nums::size_type size=locs.size();

 cout<<"\n"<<sought<<" occurs "
  <<size<<" "
  <<make_plural(size, "time", "s")<<endl;
 outfile<<"\n"<<sought<<" occurs "
  <<size<<" "
  <<make_plural(size, "time", "s")<<endl;
 line_nums::const_iterator it=locs.begin();
 for ( ;it!=locs.end();it++)
 {
  cout<<"\t(line"<<(*it)+1<<") "<<file.text_line(*it)<<endl;
  outfile<<"\t(line"<<(*it)+1<<") "<<file.text_line(*it)<<endl;
 }
  
}

主函数的实现：

int main(int argc,char **argv)  //**argv 以数组为形参
{
 ifstream infile;

 if (argc<2||!open_file(infile, argv[1]))   //argv[1]打开word.txt文件
 {
  cerr<<"No input file!"<<endl;
  return EXIT_FAILURE;
 }

 TextQuery tq;
 tq.read_file(infile);             //读取文件中的数据

 while (true)
 {
  cout<<"enter word to look for ,or q to quit:";
  string s;
  cin>>s;
  if(!cin||s=="q") break;

  set<TextQuery::line_no> locs=tq.run_query(s);

  print_result(locs,s,tq);
 }

 return 0;
}

最后别忘了加入所需的头文件以及命名空间。