64,282
社区成员
发帖
与我相关
我的任务
分享
/* text_query.h*/
#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <string>
using namespace std;
class TextQuery
{
public:
typedef vector< string >::size_type line_no;
typedef map< string,set<line_no> >::iterator map_word_iterator;
void read_file(ifstream &is)
{
store_file(is);
build_map();
}
set< line_no > run_query(const string &) const;
string text_line(line_no) const;
private:
void store_file(ifstream &);
void build_map();
vector< string > lines_of_text;
map< string,set< line_no > > word_map;
};
/*test_query.cpp*/
#pragma warning(disable:4786)
#include <iostream>
#include <fstream>
#include <sstream>
#include <vector>
#include <map>
#include <set>
#include <string>
#include "textQuery.h"
using namespace std;
void TextQuery::store_file(ifstream &is)
{
string textline;
while (getline(is, textline))
lines_of_text.push_back(textline);
}
void TextQuery::build_map()
{
for(line_no line_num = 0; line_num != lines_of_text.size(); ++line_num)
{
istringstream line(lines_of_text[line_num]);
string word;
while (line >>word)
word_map[word].insert(line_num);
}
}
set< TextQuery::line_no > TextQuery::run_query(const string &query_word) const
{
map< string, set< line_no > >::const_iterator loc = word_map.find(query_word);
if (loc == word_map.end())
return set<line_no>();
else
return loc->second;
}
string TextQuery::text_line(line_no line) const
{
if (line <lines_of_text.size())
return lines_of_text[line];
throw out_of_range("line number out of ranger");
}
/*main.cpp*/
#pragma warning(disable:4786)
#include <iostream>
#include <fstream>
#include <sstream>
#include <vector>
#include <map>
#include <set>
#include <string>
#include "textQuery.h"
using namespace std;
typedef set<TextQuery::line_no> line_nums;
void print_results(const set<TextQuery::line_no> &locs, const string &sought, const TextQuery &file);
ifstream& open_file(ifstream &in, const string &file);
int main()
{
ifstream infile;
if ( !open_file(infile, "text_in.txt") )
{
cerr << "No input file!" << endl;
return EXIT_FAILURE;
}
TextQuery tq;
tq.read_file(infile);
while (true)
{
cout << "enter word to look for,or q for qiut: ";
string s;
cin >> s;
if(!cin || s == "q")
break;
set<TextQuery::line_no> locs =tq.run_query(s);
print_results(locs, s, tq);
}
return 0;
}
void print_results(const set<TextQuery::line_no> &locs, const string &sought, const TextQuery &file)
{
line_nums::size_type size = locs.size();
cout << "\n" <<sought << " occurs " << size << " " <</*make_plural(size, "time" , "s") <<*/endl;
line_nums::const_iterator it = locs.begin();
for( ; it != locs.end(); ++it)
{
cout <<"\t(line " << (*it) +1 << ")" <<file.text_line(*it) <<endl;
}
}
ifstream& open_file(ifstream &in, const string &file)
{
in.close();
in.clear();
in.open(file.c_str());
return in;
}