文本查询程序

这将实现一个简单的文本查询程序，使用到shared_ptr智能指针，map、set、vector等容器。适合学习

#include<iostream>
#include<sstream>
#include<map>
#include<set>
#include<vector>
#include<memory>
#include<string>
#include<fstream>
using namespace std;

class QueryResult;

class TextQuery{
public:
    using line_no = vector<string>::size_type;
    TextQuery(ifstream &infile);
    QueryResult query(const string &q_text) const;

private:
    // 使用智能指针来管理动态分配的内存，当没有引用指向他的时候，他会自动解析，释放内存
    shared_ptr<vector<string>> file;// 输入文件
    map<string, shared_ptr<set<line_no>>> wm;
};

class QueryResult
{
    friend ostream& print(ostream& ,const QueryResult &);
public:
    using line_no = vector<string>::size_type;
    QueryResult(string s, shared_ptr<set<line_no>> p, shared_ptr<vector<string>> f)
    :sought(s), lines(p), file(f){ }
private:
    string sought;
    shared_ptr<set<line_no>> lines;
    shared_ptr<vector<string>> file;
};

// 1、保存text
// 2、把单词的行号映射到set中
TextQuery::TextQuery(ifstream &infile):file(new vector<string>)
{
    string text;
    int n = 0;
    // if (! infile) cout<<"No"<<endl;
    // getline(infile, text);
    // cout<<text<<endl;
    while (getline(infile, text)) {
        // cout<<"abc----"<<endl;
        file->push_back(text);
        cout<<text<<endl;
        istringstream line(text);
        string word;
        while (line >> word){
            // 如果单词不在wm中，则在wm中添加
            auto &lines = wm[word];
            if (! lines)// 如果不存在，相当于空的指针，需要为它分配内存地址
                lines.reset(new set<line_no>);
            lines->insert(n);
        }
        //get 完一行后，n++
        n ++;
    }
}

// 该函数的作用是查找sought， 然后初始化单词的所在行和出现的行号set
QueryResult TextQuery::query(const string &sought) const 
{
    // 如果找不到单词，则返回此空的set指针，但是有地址
    static shared_ptr<set<line_no>> nodata(new set<line_no>);
    auto loc = wm.find(sought);
    if (loc == wm.end())
        return QueryResult(sought, nodata, file);
    else 
        return QueryResult(sought, loc->second, file);
}

ostream& print(ostream &os, const QueryResult &qr){
    os<< qr.sought <<" occurs "<< qr.lines->size()<<" "
      << " times"<<endl;
    for (auto num : *qr.lines)
        os << "\t(line "<<num + 1<<")"<< *(qr.file->begin() + num)<<endl;
    return os;
}


void runQueries(ifstream &infile)
{
    TextQuery tq(infile);

    while (1){
        cout<<"enter a word to look for, or q to quit: ";
        string s;
        if ( !(cin >> s) || s == "q") break;
        print(cout, tq.query(s)) << endl;

    }
}
int main()
{
    ifstream infile;
    infile.open("D:\\words.txt");
    if (infile.is_open())
        runQueries(infile);
    else 
        cout<<"nonono"<<endl;
    return 0;
}