C++Primer 【笔记】文本查询程序 TextQuery

《c++ primer》容器这一部分有一个综合应用的小程序设计。要求：程序将读取用户指定的任意文本文件，然后允许用户从该文件中查找单词。查询的结果是该单词出现的次数，并列出每次出现所在的行。如果某单词在同一行中多次出现，程序将只显示该行一次。行号按升序显示，即第 7 行应该在第 9 行之前输出，依此类推。
设计程序的一个良好习惯是首先将程序所涉及的操作列出来。明确需要提供的操作有助于建立需要的数据结构和实现这些行为。从需求出发，我们的程序需要支持如下任务：
1. 它必须允许用户指明要处理的文件名字。程序将存储该文件的内容，以便输出每个单词所在的原始行。
2. 它必须将每一行分解为各个单词，并记录每个单词所在的所有行。在输出行号时，应保证以升序输出，并且不重复。
3. 对特定单词的查询将返回出现该单词的所有行的行号。
4. 输出某单词所在的行文本时， 程序必须能根据给定的行号从输入文件中获取相应的行。

我们将用一个简单的类 TextQuery 实现这个程序。再加几种容器的配合使用，就可相当巧妙地满足上述要求。
1. 使用一个 vector< string > 类型的对象存储整个输入文件的副本。输入文件的每一行是该 vector 对象的一个元素。因而，在希望输出某一行时，只需以行号为下标获取该行所在的元素即可。
2. 使用一个容器multimap将每个单词和其所在的行号记下来。
3.在查询时将每个单词所在的行号存储到一个 set 容器对象中。使用 set 就可确保每行只有一个条目，而且行号将自动按升序排列。
4. 最后Vector< string >通过set中的行号，直接利用下标读取相应行内容。
综上所述，我们定义的 TextQuery 类将有两个数据成员：储存输入文件的vector 对象，以及一个multimap 容器对象，该对象关联每个输入的单词以及记录该单词所在行号。

该类的接口需提供下列三个 public 函数：
• read_file 成员函数，其形参为一个 ifstream& 类型对象。该函数每次
从文件中读入一行，并将它保存在 vector 容器中。输入完毕后，
read_file 将创建关联每个单词及其所在行号的 multimap 容器。
• run_query 成员函数， 其形参为一个 string 类型对象， 返回一个 set 对象，该 set 对象包含出现该 string 对象的所有行的行号。
• text_line 成员函数，其形参为一个行号，返回输入文本中该行号对应的
文本行。
程序如下：

#include<iostream>
#include<fstream>
#include<vector>
#include<map>
#include<sstream>
#include<set>

using namespace std;

class TextQuery{
    private:
        vector<string> lines; //store files by lines
        multimap<string, int> word_with_line_number; //store every word and its line number
    public: 
        typedef vector<string>::size_type sz_line;
        typedef multimap<string, int>::iterator multi_iter;
    //read files into a vector by line and insert every word and its line number into a map
    void read_file(ifstream &text){
        string line;
        sz_line line_number = 0;//record the line number
        while(getline(text, line)){//read files in lines
            lines.push_back(line);
            istringstream stream(line);
            string word;
            while(stream >> word)// read line in words
                word_with_line_number.insert(make_pair(word, line_number));
            ++line_number;//add line number after reading a line
        }
    }

    //run the query and return the word's lines number in a set
    set< sz_line > run_query(string search_word)  {
        pair<multi_iter, multi_iter> pos =  word_with_line_number.equal_range(search_word);
        set< sz_line > word_lines;//all lines' numbers for a word
        while(pos.first != pos.second){
            word_lines.insert(pos.first->second);//store the lines' number in the set
            ++pos.first;
        }
        return word_lines;
    }

    //get the text line with certain line number
    string text_line(sz_line line_number)  {
        return lines[line_number];//read lines by index
    }

};


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

//put out the results
void print_results(set< TextQuery::sz_line > locations, string s, TextQuery tq){
    if(locations.size() == 0)
        cout << "There is no " << s <<endl;
    else{
        cout << "Element occurs " << locations.size() << " times:" << endl;
        for(set< TextQuery::sz_line >::iterator iter = locations.begin(); iter != locations.end(); ++iter){
            //get the line number from the set and then call the text_line function to read contents
            cout << "(Line " << *iter + 1  << ") " << tq.text_line(*iter) << endl;//the line read from 0, so add 1 in the output
        }
    }
}

int main(int argc, char *argv[]){
    ifstream infile;
    if(argc < 2 || !open_file(infile, argv[1])){
        cerr << "No input file!" << endl;

    }
    TextQuery tq;
    tq.read_file(infile);
    while(true){
        cout << "enter word  you want to search or 'q' toquit:  ";
        string s;
        cin >> s;
        if(s == "q" || !cin)
            break;
        set<TextQuery::sz_line> locations = tq.run_query(s);//run query get line numbers
        print_results(locations, s, tq);// put out results
        cout << endl;
    }
    return 0;
}