C/C++:文本查询(单词查询)

如题:

C/C++:

Textqurey.h(方便看都在.h里实现了):

 //
 // Created by 徐爱东 on 17/7/10.
 //

 #ifndef TEXTQUERY_TEXTQUERY_H
 #define TEXTQUERY_TEXTQUERY_H

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

 class Textquery
 {
 private:
     using vsizetype=std::vector<std::string>::size_type;
     std::shared_ptr<std::vector<std::string> > File;
     std::map<std::string, std::shared_ptr<std::set<vsizetype> > > wm;

 public:
     class QueryResult
     {
         friend class Textquery;

     private:
         std::string Queryword;
         std::shared_ptr<std::vector<std::string> > file_;
         std::shared_ptr<std::set<vsizetype> > lines_;
     public:
         QueryResult(std::string Queryword, std::shared_ptr<std::vector<std::string> > file,
                     std::shared_ptr<std::set<vsizetype> > lines)
                 : Queryword(Queryword),
                   file_(file),
                   lines_(lines)
         {

         }
     };

     Textquery(std::fstream &infile)
             : File(new std::vector<std::string>)
     {
         //文本存入vector并计算单词行数入set
         std::string Textline;
         std::stringstream stream;
         while (std::getline(infile, Textline))
         {
             stream.clear();          //标志位清理
             File->push_back(Textline);
             stream << Textline;
             vsizetype linenumber = File->size() - ; //行号;

             while (stream >> Textline)          //单词根据空格分隔
             {
                 auto &lines = wm[Textline];     //map[]方式没有则创建，有则返回value_type;
                 if (!lines)                     //指针为空，则创建
                     lines.reset(new std::set<vsizetype>);
                 lines->insert(linenumber);
             }
         }
     }

     QueryResult query(const std::string &str)
     {
         auto loca=wm.find(str);

         if ( loca!= wm.end())
         {
             return QueryResult(str, File, wm[str]);
         } else
         {
             std::shared_ptr<std::set<vsizetype>> shared_null(new std::set<vsizetype >);    //没找到则返回空指针,防止崩溃
             return QueryResult(str, File, shared_null);
         }
     }

     std::ostream &display(std::ostream &out, QueryResult result)
     {
         if (!result.lines_)
         {
             return out << result.Queryword << " appear 0 time\n";

         } else
         {
             out << result.Queryword << " appear " << result.lines_->size()
                 << (result.lines_->size() >  ? " times" : " time") << "\n";

             for (auto &iter:*result.lines_)
             {
                 out << "\t\t\t\t\t(line" << iter +  << "): " << *(result.file_->begin() + iter) << std::endl;
             }

             //不能- -for_each, lamda语法报错？
 //            std::for_each(result.lines_->begin(), result.lines_->end(), [&](vsizetype &linenumber)
 //            {
 //                out<<"\t(line"<<linenumber+1<<"): "<<*(result.file_->begin()+linenumber)<<std::endl;
 //            });

             return out;
         }
     }
 };

 #endif //TEXTQUERY_TEXTQUERY_H

main.cpp:

#include <iostream>
#include "Textquery.h"

int main()
{
    std::fstream file("./Tex.txt", std::ios::in);
    std::stringstream stream;

    Textquery query(file);
    std::string word;
    do
    {
        std::cout << "Please input search word or q to quit: ";
        std::cin >> word;

        if (word == "q")break;

        query.query(word);
        query.display(std::cout, query.query(word));

    } while ();
    std::cout<<"end"<<std::endl;
    return ;
}

运行结果: