文本查询程序

最新推荐文章于 2022-05-15 12:24:52 发布

hjhomw

最新推荐文章于 2022-05-15 12:24:52 发布

阅读量410

点赞数

分类专栏： C++primer 文章标签： c-c++

本文链接：https://blog.csdn.net/hejun_haitao/article/details/48261389

版权

C++primer 专栏收录该内容

107 篇文章 1 订阅

订阅专栏

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

using line_no=vector<string>::size_type;


class QueryResult;
class TextQuery
{
public:

    TextQuery(ifstream&);
    QueryResult query(const string&) const;

private:
    shared_ptr<vector<string>> file;
    map<string,shared_ptr<set<line_no>>> wm;
};

TextQuery::TextQuery(ifstream& is):file(new vector<string>)
{
    string text;
    while(getline(is,text))
    {
        file->push_back(text);
        int n=file->size()-1;
        istringstream line(text);
        string word;
        while(line>>word)
        {
            auto &lines=wm[word];
            if(!lines)
                lines.reset(new set<line_no>);
            lines->insert(n);
        }

    }
}


class QueryResult
{

public:
    friend ostream& print(ostream&,const QueryResult&);
    friend string make_plural(size_t ctr,const string &word,const string &ending);
    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;
};

QueryResult TextQuery::query(const string&sought) const
{
    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);
}

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

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

    return os;

}


int main()
{
    ifstream infile("TextQuery.text");
    TextQuery tq(infile);
    while(true)
    {
        cout<<"enter word to look for,or q to quit: "<<endl;
        string s;
        if(!(cin>>s) || s=="q")  break;
        print(cout,tq.query(s))<<endl;
    }


    return 0;
}