Query查询练习，使用map vector

/*************************************************************************
    > File Name: Query.h
    > Author:keson
    > Mail:keson@bupt.edu.cn
    > Created Time: 2014年11月05日 星期三 16时46分18秒
 ************************************************************************/

#ifndef _QUERY_H
#define _QUERY_H
#include<fstream>
#include<vector>
#include<string>
#include<sstream>
#include<map>
#include<set>
using namespace std;
class TextQuery
{
    public:
    TextQuery(std::ifstream &);            //构造函数
    
    TextQuery read(ifstream &file,TextQuery &item);  //读输入文件
    void print();    //打印读入的文件
    void account(ifstream &file);   //根据输入流将key值和value统计输入map
    void query(string &s);          //根据单词s打印出所在行
    void map_print();               //打印map的key值
    private:
    vector<string> vec;             //用来保存输入的每一行
    map<string,set<int>> word_map;  //每个单词作为key,行号的set集作为value

};
void TextQuery::map_print()         //打印出现的单词
{
    auto ret=word_map.begin();
    while(ret!=word_map.cend())
    {
        cout<<ret->first<<" ";
        ++ret;
    }
}
void TextQuery::query(string &s)         //根据输入的单词s，打印出所在行
{
    auto ret=word_map.find(s);
    if(ret!=word_map.cend())
    {
        auto set_it=(ret->second).cbegin();
        while(set_it!=(ret->second).cend())
        {
            cout<<*set_it<<"   "<<vec[*set_it-1]<<endl;
            ++set_it;
        }
    }
    else
    cout<<"could't find";
}
void TextQuery::account(ifstream &file)
{
    string fline,word;
    int line_num=0;
    while(getline(file,fline))
    {
        ++line_num;
        istringstream record(fline);   //将一行绑定到record流，分解成单词
        while(record>>word)
        {
            auto ret=word_map.find(word);
            if (ret!=word_map.cend())          //若该单词已经在map的key值出现，则插入行号
            {
             ret->second.insert(line_num);
            }
            else                               //若未出现，则插入key和对应的value
            {
             set<int> val={line_num};
             word_map.insert(make_pair(word,val));
            }   
        }
        
    }
}
void TextQuery::print()         //打印出vec中的内容
{
    for(auto c:vec)
        cout<<c<<endl;
}

TextQuery TextQuery::read(std::ifstream &file,TextQuery &item)
{

    string line;
    if(file)   //如果打开成功，读入每一行
    {
        while(getline(file,line))
        {
            (item.vec).push_back(line);
        }

    }
    else
      cerr<<"could't open";
    
    file.close();

    return item;
}

TextQuery::TextQuery(std::ifstream &file)
{
    read(file,*this);   //read读入初始化vec
    map<string,set<int>> word_map;

}
#endif

/*************************************************************************
    > File Name: Query.cpp
    > Author:keson
    > Mail:keson@bupt.edu.cn
    > Created Time: 2014年11月05日 星期三 16时38分07秒
 ************************************************************************/

#include<iostream>
#include<fstream>
#include"Query.h"
using namespace std;

int main()
{
    ifstream file("storyDataFile");
    TextQuery item(file);
    item.print();
    file.close();                     //关闭file流
    ifstream file2("storyDataFile");  //重新打开那个文件
    item.account(file2);
    cout<<"Enter the word you want to find "<<endl;
    string s;
    cin>>s;
    item.query(s);
   // item.map_print();
    return 0;
    
}
/*
void runQueries(ifstream &infile);


void runQueries(ifstream &infile)
{
    //infile是一个ifstream,指向我们要处理的文件
    Text tq(infile);  //保存建立并打印结果

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

}
*/