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() - 1; //行号;

            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
";

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


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

            //不能- -for_each, lamda语法报错？
//            std::for_each(result.lines_->begin(), result.lines_->end(), [&](vsizetype &linenumber)
//            {
//                out<<"	(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 (1);
    std::cout<<"end"<<std::endl;
    return 0;
}

运行结果: