C++ Primer 关联容器--容器的综合应用：文本查询程序

//TextQuery.h

#ifndef _TEXTQUERY_H_

#define _TEXTQUERY_H_

#include <vector>

#include <fstream>

#include <iostream>

#include <string>

#include <set>

#include <map>

class TextQuery{

public:

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;

};

#endif

// textQuery.cpp : 定义控制台应用程序的入口点。

#include "stdafx.h"

#include "textquery.h"

#include <iostream>

#include <sstream>

#include <utility>

using std::string;

using std::vector;

using std::set;

using std::map;

using std::istringstream;

std::set<TextQuery::line_no> TextQuery::run_query(const std::string &word) const

{

map< string, set<line_no> >::const_iterator iter = word_map.find(word);

if (iter != word_map.end())

{

return iter->second;

}

else

{

set<line_no> temp;

return temp;

//return set<line_no>();//书上使用这种方法，简明了许多

}

}

std::string TextQuery::text_line(line_no linenumber) const

{

return lines_of_text[linenumber];

}

void TextQuery::store_file(std::ifstream &is)

{

string line;

while (getline(is, line))

{

lines_of_text.push_back(line);

}

}

void TextQuery::build_map()

{

string word;

map< string, set<line_no> >::iterator iter;

for (line_no ix = 0; ix != lines_of_text.size(); ++ix)

{

istringstream stream(lines_of_text[ix]);

//istringstream stream;

//stream.str(lines_of_text[ix]);//有两种将string转换为istringstream的方法

while (stream >> word)

{

//word_map[word].insert(ix);//书上用了这种写法，只需一句简单明了多了

iter = word_map.find(word);

if (iter != word_map.end())

{

iter->second.insert(ix);

}

else

{

set<line_no> lineset;

lineset.insert(ix);

word_map.insert(make_pair(word, lineset));

}

}

}

}

//test.cpp

#include"stdafx.h"

#include "textquery.h"

#include <iostream>

#include <string>

#include <set>

#include <fstream>

#include <utility>

using std::ifstream;

using std::cerr;

using std::cout;

using std::cin;

using std::string;

using std::endl;

using std::set;

using std::runtime_error;

ifstream& open_file(ifstream& in, const string& file)

{

in.close();

in.clear();

in.open(file.c_str());

return in;

}

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 << "\n" << sought << " occurs "

<< size << " "

<< "times" << endl;

line_nums::const_iterator iter = locs.begin();

for ( ; iter != locs.end(); ++iter)

{

cout << "\t(line "

<< (*iter) + 1 << ") "

<< file.text_line(*iter) << endl;

}

}

int main(int argc, char **argv)

{

/*下面两种文件的读取方式均可。运行程序有两种方法

方法一：利用cmd命令到命令行窗口，进入到可执行文件所在的目录

输入xxx.exe yyy.txt

方法二：利用vs2008属性-调试-命令参数中填写yyy.txt

然后在工作目录中填写可执行程序所在目录，将yyy.txt也放入此目录中

要注意的是对于win7系统，文件名后缀是不显示的，所以在命名文件时

不需要写yyy.txt，这样会出现文件全名变成yyy.txt.txt的情况*/

#if 0

ifstream infile;

if (argc < 2 || !open_file(infile, argv[1]))

{

cerr << "No input file!" << endl;

//return EXIT_FAILURE;

}

//open_file(infile, argv[1]);

#endif

#if 1

if (argc != 2)

{

throw runtime_error("wrong number of arguments");

}

ifstream infile;

if (!open_file(infile, argv[1]))

{

throw runtime_error("no input file");

}

#endif

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_results(locs, s, tq);

}

return 0;

}