只定义了一个类,代码感觉要精简一点
类定义:CTextQuery.h
#pragma once
#include <string>
#include <vector>
#include <map>
#include <set>
#include <sstream>
#include <fstream>
#include <iostream>
class TextQuery {
public:
TextQuery(std::ifstream& infile);
void seachAndPrint(std::string& word);//查询打印结果
private:
//保存每一行的vector,行号从0开始
std::vector<std::string> vtext{};
//key:单词 value:出现的行的集合
std::map<std::string, std::set<size_t>> word_line{};
//查询:map.size()为出现的次数,map.first为每一个行号,second为对应行的文本
std::map<size_t, std::string> sarchWord(std::string& word);
};
inline TextQuery::TextQuery(std::ifstream& infile)
{
std::string line;
size_t sz = 0;
while (getline(infile, line))
{
vtext.push_back(line);//读取每一行保存到vector
std::istringstream is_line(line);
std::string word;
while (is_line >> word)
{
word_line[word].insert(sz);//保存word到set<size_t>的map
}
++sz;
}
}
inline std::map<size_t, std::string> TextQuery:: sarchWord(std::string& word)
{
std::map<size_t, std::string> ret{};
auto iter = word_line.find(word);
if (iter != word_line.end())
{
auto it = vtext.begin();
for (auto &i : iter->second)//提取每一个行号
{
ret.insert({ i,*(it +i) });//插入元素
}
}
return ret;
}
inline void TextQuery::seachAndPrint(std::string& word)
{
auto result =sarchWord(word);
std::cout << word << "一共出现" << result.size() << "次" << std::endl;
if (result.size() != 0)
{
for (auto &w : result)
{
std::cout << "第 " << w.first + 1 << " 行:" << w.second << std::endl;
}
}
}
main.cpp
#include "stdafx.h"
#include "CTextQuery.h"
using namespace std;
int main()
{
ifstream infile("input.txt");
TextQuery mquery(infile);
string word;
cin >> word;
mquery.seachAndPrint(word);
getchar();
getchar();
return 0;
}